Skip to content
Snippets Groups Projects
Commit 92ab889c authored by Daniel Suo's avatar Daniel Suo Committed by TensorFlower Gardener
Browse files

Implement Jax CPU/GPU callbacks with XLA's FFI. 

- Change 4 of 4 addressing #3 in https://github.com/jax-ml/jax/issues/25842.

PiperOrigin-RevId: 720696248
parent 44dc41df
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 371 additions and 18 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment