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

Fix circular import in pallas core file (#354)

Browse Source
This commit is contained in:
Ruturaj Vaidya 2025-04-10 10:33:25 -05:00 committed by GitHub
parent ac21549df0
commit 5557c1d642
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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 util
from jax._src.pallas import core as pallas_core
from jax._src.pallas import pallas_call
import jax.numpy as jnp
import numpy as np