* [GitHub commits](https://github.com/google/jax/compare/jax-v0.2.0...jax-v0.2.1). * Improvements: * As a benefit of omnistaging, the host_callback functions are executed (in program order) even if the result of the {py:func}`jax.experimental.host_callback.id_print`/ {py:func}`jax.experimental.host_callback.id_tap` is not used in the computation.
0.2.0
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.1.77...jax-v0.2.0). * Improvements: * Omnistaging on by default. See {jax-issue}`3370` and [omnistaging](https://github.com/google/jax/blob/main/docs/design_notes/omnistaging.md)
0.1.77
* Breaking changes: * New simplified interface for {py:func}`jax.experimental.host_callback.id_tap` (4101)
jaxlib 0.1.55 (September 8, 2020)
* Update XLA: * Fix bug in DLPackManagedTensorToBuffer (4196)