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-24 15:46:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax_plugins
/
cuda
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
..
__init__.py
[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
plugin_pyproject.toml
[PJRT C API] Rename the folder "plugins" to "jax_plugins".
2023-11-14 13:56:13 -08:00
plugin_setup.py
[PJRT C API] Rename the folder "plugins" to "jax_plugins".
2023-11-14 13:56:13 -08:00
pyproject.toml
[PJRT C API] Rename the folder "plugins" to "jax_plugins".
2023-11-14 13:56:13 -08:00
setup.py
[PJRT C API] Rename the folder "plugins" to "jax_plugins".
2023-11-14 13:56:13 -08:00
