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 authored and Google-ML-Automation committed Oct 5, 2024
1 parent 45f0e9a commit 9be9b67
Show file tree
Hide file tree
Showing 6 changed files with 40 additions and 4,056 deletions.
3 changes: 3 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,9 @@ When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.
* {func}`jax.numpy.isscalar` now returns True for any array-like object with
zero dimensions. Previously it only returned True for zero-dimensional
array-like objects with a weak dtype.
* `jax.experimental.host_callback` has been deprecated since March 2024, with
JAX version 0.4.26. Now we removed it.
See {jax-issue}`#20385` for a discussion of alternatives.

## jax 0.4.34 (October 4, 2023)

Expand Down
Loading

0 comments on commit 9be9b67

Please sign in to comment.