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

Commit

1 year 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