Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[host_callback] Remove most of the jax.experimental.host_callback module
These APIs have been deprecated since March 2024 and they are subsumed by the new JAX external callbacks. See #20385 for a discussion. PiperOrigin-RevId: 682659677
- Loading branch information