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

Fix circular import in pallas core file

Browse Source
This commit is contained in:
Ruturaj4 2025-04-09 19:36:15 -05:00
parent ac21549df0
commit ce7347a52b
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/_src/pallas/mosaic/core.py
View File

@ -28,7 +28,6 @@ from jax._src import core as jax_core
from jax._src import dtypes from jax._src import dtypes
from jax._src import util from jax._src import util
from jax._src.pallas import core as pallas_core from jax._src.pallas import core as pallas_core
from jax._src.pallas import pallas_call
import jax.numpy as jnp import jax.numpy as jnp
import numpy as np import numpy as np