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-19 05:16:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jaxlib
/
mlir
/
_mlir_libs
History
Peter Hawkins
9c3365fb95
Migrate shardy dialect extension to nanobind.
...
PiperOrigin-RevId: 707991933
2024-12-19 12:25:14 -08:00
..
_triton_ext.pyi
[jaxlib] Added a missing pytype_dep to :_triton_ext
2024-12-19 07:28:28 -08:00
BUILD.bazel
Migrate shardy dialect extension to nanobind.
2024-12-19 12:25:14 -08:00
mosaic_gpu_ext.cc
Migrate JAX MLIR Python dialect extensions to nanobind.
2024-12-13 07:08:28 -08:00
register_jax_dialects.cc
Migrate JAX MLIR Python dialect extensions to nanobind.
2024-12-13 07:08:28 -08:00
tpu_ext.cc
Add Python.h include to fix Windows build.
2024-12-16 07:21:01 -08:00
triton_ext.cc
Migrate JAX MLIR Python dialect extensions to nanobind.
2024-12-13 07:08:28 -08:00
