mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 00:56:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/lib/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
Switch Mosaic GPU to a custom pass pipeline and improve the lowering of GPU launch
2024-04-24 03:27:45 -07:00