jax
28afd252 - Add FFI example demonstrating the use of XLA's FFI state.

Commit

331 days ago

Add FFI example demonstrating the use of XLA's FFI state. Support for this was added in JAX v0.5.0. PiperOrigin-RevId: 722597704

References

Author

dfm

Committer

Google-ML-Automation

Parents