This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-17 04:16:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
lib
History
Benjamin Chetioui
837418c652
[Mosaic GPU] Remove old jaxlib version guards.
...
PiperOrigin-RevId: 726071956
2025-02-12 08:49:40 -08:00
..
mlir
Partially rolling forward
#22998
2024-09-02 04:44:47 -07:00
__init__.py
[Mosaic GPU] Remove old jaxlib version guards.
2025-02-12 08:49:40 -08:00
BUILD
[Mosaic GPU] Add Python bindings for the Mosaic GPU MLIR dialect.
2024-10-31 02:47:30 -07:00
mosaic_gpu.py
[mosaic_gpu] Fixed mosaic_gpu-serde pass registration
2025-01-24 06:35:54 -08:00
triton.py
[pallas:triton] The lowering now uses PTX instead of Triton IR
2025-02-03 13:21:40 -08:00
