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
29f1d3b033
[PJRT C API] Use xla_client.generate_pjrt_gpu_plugin_options to generate options for CUDA plugin.
...
PiperOrigin-RevId: 603074180
2024-01-31 09:42:58 -08:00
..
cuda
[PJRT C API] Use xla_client.generate_pjrt_gpu_plugin_options to generate options for CUDA plugin.
2024-01-31 09:42:58 -08:00
BUILD.bazel
Support bazel test without bazel build for CUDA PJRT plugin.
2023-11-15 10:38:19 -08:00
