Skip to content

Commit

Permalink
[host_callback] Remove most of the jax.experimental.host_callback module
Browse files Browse the repository at this point in the history 
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
@gnecula @Google-ML-Automation
gnecula authored and Google-ML-Automation committed Oct 5, 2024
1 parent 45f0e9a commit ac42ebb
Show file tree
Hide file tree
Showing 4 changed files with 24 additions and 4,016 deletions.
Loading

0 comments on commit ac42ebb

Please sign in to comment.