mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax_plugins/rocm
History
Dan Foreman-Mackey a7f384cc6e Add a register_custom_type_id function to the GPU plugins. 
This enables dynamic registration of custom FFI types on the appropriate platform via PJRT.

PiperOrigin-RevId: 712904085
2025-01-07 07:29:38 -08:00
..
__init__.py
Add a register_custom_type_id function to the GPU plugins.
2025-01-07 07:29:38 -08:00
BUILD.bazel
Replace pjrt_c_api_gpu_plugin.so symlink with XLA dependency.
2024-08-12 13:01:18 -07:00
plugin_pyproject.toml
[ROCm] Bring-up pjrt support
2024-06-17 16:49:22 +00:00
plugin_setup.py
[ROCm] Add rocm version information
2024-11-25 10:21:48 -06:00
pyproject.toml
[ROCm] Bring-up pjrt support
2024-06-17 16:49:22 +00:00
setup.py
[ROCm] Add rocm version information
2024-11-25 10:21:48 -06:00