mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
Muhammad Abdullah 2023-08-28 10:13:16 +05:00 committed by GitHub
parent d09c55aa2d
commit 599b35eeaa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/__init__.py
View File

@ -156,6 +156,7 @@ from jax import custom_transpose as custom_transpose
from jax import api_util as api_util
from jax import distributed as distributed
from jax import debug as debug
from jax import dlpack as dlpack
from jax import dtypes as dtypes
from jax import errors as errors
from jax import image as image
@ -172,7 +173,6 @@ from jax import sharding as sharding
from jax import stages as stages
from jax import tree_util as tree_util
from jax import util as util
from jax import dlpack as dlpack
# Also circular dependency.
from jax._src.array import Shard as Shard