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-16 11:56:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jaxlib
/
mlir
/
_mlir_libs
History
Adam Paszke
99a12ef9ea
[Mosaic GPU] Add support for warpgroup lowering of loops with vector carries
...
PiperOrigin-RevId: 731260912
2025-02-26 04:29:36 -08:00
..
_triton_ext.pyi
[jaxlib] Added a missing pytype_dep to :_triton_ext
2024-12-19 07:28:28 -08:00
BUILD.bazel
[mosaic_gpu] Fixed mosaic_gpu-serde pass registration
2025-01-24 06:35:54 -08:00
mosaic_gpu_ext.cc
[Mosaic GPU] Add support for warpgroup lowering of loops with vector carries
2025-02-26 04:29:36 -08:00
register_jax_dialects.cc
[mosaic_gpu] Fixed mosaic_gpu-serde pass registration
2025-01-24 06:35:54 -08:00
tpu_ext.cc
[Mosaic] Remove hardcoded TARGET_SHAPE and align Python/C++ APIs.
2025-01-21 10:24:10 -08:00
triton_ext.cc
Migrate JAX MLIR Python dialect extensions to nanobind.
2024-12-13 07:08:28 -08:00
