mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/experimental/mosaic
History
Benjamin Chetioui 837418c652 [Mosaic GPU] Remove old jaxlib version guards. 
PiperOrigin-RevId: 726071956
2025-02-12 08:49:40 -08:00
..
gpu
[Mosaic GPU] Remove old jaxlib version guards.
2025-02-12 08:49:40 -08:00
__init__.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
dialects.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00