mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 00:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/extend/mlir
History
Vladimir Belitskiy 5370ac2ec5 Remove the try/except for Shardy imports. 
Shardy has been been included in JAX for a while now.

PiperOrigin-RevId: 742778405
2025-04-01 11:33:44 -07:00
..
dialects
Remove the try/except for Shardy imports.
2025-04-01 11:33:44 -07:00
__init__.py
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
BUILD
Create jax wheel build target.
2025-02-25 09:30:08 -08:00
ir.py
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
passmanager.py
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00