Jax

* 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)

* [GitHub commits](https://github.com/google/jax/compare/jax-v0.1.75...jax-v0.1.76).

* [GitHub commits](https://github.com/google/jax/compare/jax-v0.1.74...jax-v0.1.75).
* Bug Fixes:
* make jnp.abs() work for unsigned inputs (3914)
* Improvements:
* "Omnistaging" behavior added behind a flag, disabled by default (3370)

* [GitHub commits](https://github.com/google/jax/compare/jax-v0.1.73...jax-v0.1.74).
* New Features:
* BFGS (3101)
* TPU support for half-precision arithmetic (3878)
* Bug Fixes:
* Prevent some accidental dtype warnings (3874)
* Fix a multi-threading bug in custom derivatives (3845, 3869)
* Improvements:
* Faster searchsorted implementation (3873)
* Better test coverage for jax.numpy sorting algorithms (3836)

jaxlib 0.1.52 (July 22, 2020)

* Update XLA.

* [GitHub commits](https://github.com/google/jax/compare/jax-v0.1.72...jax-v0.1.73).
* The minimum jaxlib version is now 0.1.51.
* New Features:
* jax.image.resize. (3703)
* hfft and ihfft (3664)
* jax.numpy.intersect1d (3726)
* jax.numpy.lexsort (3812)
* `lax.scan` and the `scan` primitive support an `unroll`
parameter for loop unrolling when lowering to XLA
({jax-issue}`3738`).
* Bug Fixes:
* Fix reduction repeated axis error (3618)
* Fix shape rule for lax.pad for input dimensions of size 0. (3608)
* make psum transpose handle zero cotangents (3653)
* Fix shape error when taking JVP of reduce-prod over size 0 axis. (3729)
* Support differentiation through jax.lax.all_to_all (3733)
* address nan issue in jax.scipy.special.zeta (3777)
* Improvements:
* Many improvements to jax2tf
* Reimplement argmin/argmax using a single pass variadic reduction. (3611)
* Enable XLA SPMD partitioning by default. (3151)
* Add support for 0d transpose convolution (3643)
* Make LU gradient work for low-rank matrices (3610)
* support multiple_results and custom JVPs in jet (3657)
* Generalize reduce-window padding to support (lo, hi) pairs. (3728)
* Implement complex convolutions on CPU and GPU. (3735)
* Make jnp.take work for empty slices of empty arrays. (3751)
* Relax dimension ordering rules for dot_general. (3778)
* Enable buffer donation for GPU. (3800)
* Add support for base dilation and window dilation to reduce window op… (3803)

jaxlib 0.1.51 (July 2, 2020)

* Update XLA.
* Add new runtime support for host_callback.

* [GitHub commits](https://github.com/google/jax/compare/jax-v0.1.71...jax-v0.1.72).
* Bug fixes:
* Fix an odeint bug introduced in the previous release, see
{jax-issue}`3587`.