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 6, 2024
1 parent 81a31f6 commit 4716b98
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 4716b98

Please sign in to comment.