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 20:36:05 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax_plugins
History
Jieying Luo
d6c5910105
[PJRT C API] Move cuda_plugin_extension from jaxlib to jax-cuda-plugin (the package for cuda kernels).
...
PiperOrigin-RevId: 583406466
2023-11-17 09:11:46 -08:00
..
cuda
[PJRT C API] Move cuda_plugin_extension from jaxlib to jax-cuda-plugin (the package for cuda kernels).
2023-11-17 09:11:46 -08:00
BUILD.bazel
Support bazel test without bazel build for CUDA PJRT plugin.
2023-11-15 10:38:19 -08:00
