mirrors/rocm_jax
1
0
Fork 0
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
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
..
cuda
Add a register_custom_type_id function to the GPU plugins.
2025-01-07 07:29:38 -08:00
rocm
Add a register_custom_type_id function to the GPU plugins.
2025-01-07 07:29:38 -08:00
BUILD.bazel
Don't pass --nocheck_visibility to Bazel.
2024-09-27 11:19:42 -04:00