jax
a7f384cc - Add a register_custom_type_id function to the GPU plugins.

Commit

1 year ago

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

References

#25413 - Add a register_custom_type_id function to the GPU plugins.

#194 - CI: 01/07/25 upstream sync

Author

dfm

Committer

Google-ML-Automation

Parents

853af560

jax a7f384cc - Add a register_custom_type_id function to the GPU plugins.

jax
a7f384cc - Add a register_custom_type_id function to the GPU plugins.