**It is the latest version compatible with Python 3.6. See [deepmind/optax222](https://github.com/deepmind/optax/issues/222) for more details.**
[Full Changelog](https://github.com/deepmind/distrax/compare/v0.0.2...HEAD)
**Closed issues:**
- ImportError: cannot import name 'partial' from 'jax.util' [\60](https://github.com/deepmind/distrax/issues/60)
- Problem with shapes in simple transformed distributions [\42](https://github.com/deepmind/distrax/issues/42)
- Issue with jax version in requirements.txt [\38](https://github.com/deepmind/distrax/issues/38)
- Duplication/Forking vs Collaboration [\35](https://github.com/deepmind/distrax/issues/35)
- Tanh numerical instability [\7](https://github.com/deepmind/distrax/issues/7)
**Merged pull requests:**
- Temporarily disable py3.9 tests. [\69](https://github.com/deepmind/distrax/pull/69) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Minor updates to docstring of bijectors. [\68](https://github.com/deepmind/distrax/pull/68) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Minor updates to slicing tests. [\67](https://github.com/deepmind/distrax/pull/67) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Add note for `Tanh` and `Sigmoid`. [\66](https://github.com/deepmind/distrax/pull/66) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Support slicing on more distributions. [\65](https://github.com/deepmind/distrax/pull/65) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- \[JAX\] Update JAX users in preparation for a change that makes iteration over a JAX array return JAX arrays, instead of NumPy arrays. [\63](https://github.com/deepmind/distrax/pull/63) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Fix typo in `PRNGKey`. [\62](https://github.com/deepmind/distrax/pull/62) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Fix import errors & freeze the latest compatible JAX version. Fixes \60 [\61](https://github.com/deepmind/distrax/pull/61) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Add slice to more distrax distributions. [\59](https://github.com/deepmind/distrax/pull/59) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Add slice operation to Distrax Distribution and implement for TFP, MVN and Categorical. [\58](https://github.com/deepmind/distrax/pull/58) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Update tests.yml workflow. [\57](https://github.com/deepmind/distrax/pull/57) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Add test.sh for launching CI tests on a local machine. [\56](https://github.com/deepmind/distrax/pull/56) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Use TFP nightly builds in CI tests. [\55](https://github.com/deepmind/distrax/pull/55) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Fix type of `event_shape_tensor` in Distrax-to-TFP distribution adapter. [\54](https://github.com/deepmind/distrax/pull/54) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Uncomment test cases. [\53](https://github.com/deepmind/distrax/pull/53) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Fix breakage in the TFP-to-Distrax distribution adapter. [\52](https://github.com/deepmind/distrax/pull/52) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Disable `use-list-literal` refactoring lint tests. [\51](https://github.com/deepmind/distrax/pull/51) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Remove `use-dict-literal` from the set of refactoring lint tests. [\50](https://github.com/deepmind/distrax/pull/50) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Adjust tolerance locally to fix failing test. [\48](https://github.com/deepmind/distrax/pull/48) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Change explicit string formats to f-strings to mitigate github pylint failures. [\47](https://github.com/deepmind/distrax/pull/47) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Replace uses of jax.partial and jax.util.partial with functools.partial, in preparation for removing jax.partial and jax.util.partial. [\46](https://github.com/deepmind/distrax/pull/46) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Remove users of jax.api.\* symbols, in preparation for removing the deprecated jax.api name. [\44](https://github.com/deepmind/distrax/pull/44) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Implement a numerically stable version of the quadratic formula used in the rational quadratic spline bijector. [\43](https://github.com/deepmind/distrax/pull/43) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Distrax: Enable Categorical to be used with a single element. [\41](https://github.com/deepmind/distrax/pull/41) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- Replace use of deprecated jax test utility [\40](https://github.com/deepmind/distrax/pull/40) ([copybara-service[bot]](https://github.com/apps/copybara-service))
- initial gamma distribution [\8](https://github.com/deepmind/distrax/pull/8) ([kashif](https://github.com/kashif))
\* *This Changelog was automatically generated by [github_changelog_generator](https://github.com/github-changelog-generator/github-changelog-generator)*