What's Changed
* Adjust tolerance in `gumbel_test.py`. by copybara-service in https://github.com/deepmind/distrax/pull/135
* Add an abstract class for linear bijectors. by copybara-service in https://github.com/deepmind/distrax/pull/136
* Expose linear bijectors, shift bijector, and general multivariate normal. by copybara-service in https://github.com/deepmind/distrax/pull/138
* Add missing symbols in __all__ and put list in alphabetical order. by copybara-service in https://github.com/deepmind/distrax/pull/139
* Added `log_cdf` method for the Gamma distribution and modified some tests in `gamma_test.py`. by copybara-service in https://github.com/deepmind/distrax/pull/141
* Fix `test_convert_seed`. by copybara-service in https://github.com/deepmind/distrax/pull/142
* Implement the `Beta` distribution. by copybara-service in https://github.com/deepmind/distrax/pull/140
* Migrate RLax squashed gaussian to use Distrax. Explicitly broadcast shapes in Distrax scalar affine to avoid rank promotion errors. by copybara-service in https://github.com/deepmind/distrax/pull/143
* Remove randomness in `laplace_test.py`. by copybara-service in https://github.com/deepmind/distrax/pull/144
* Implement the Dirichlet distribution. by copybara-service in https://github.com/deepmind/distrax/pull/146
* Add survival and log-survival function to `distrax.Distribution` base class. by copybara-service in https://github.com/deepmind/distrax/pull/148
* Implement survival and log-survival function for the `Logistic` distribution. by copybara-service in https://github.com/deepmind/distrax/pull/150
* Implement survival and log-survival function for the `Normal` distribution. by copybara-service in https://github.com/deepmind/distrax/pull/151
* Raise `ValueError` when input parameters are not valid for `Bernoulli`, `Multinomial`, and `Categorical`. by copybara-service in https://github.com/deepmind/distrax/pull/153
* Remove `RTOL` from all distribution tests. by copybara-service in https://github.com/deepmind/distrax/pull/152
* Documentation changes to `LogStddevNormal`. by copybara-service in https://github.com/deepmind/distrax/pull/154
* Improve numerical stability of prob/log_prob computation of `distributions.Quantized`. by copybara-service in https://github.com/deepmind/distrax/pull/149
* Improve the tests in `beta_test.py`. by copybara-service in https://github.com/deepmind/distrax/pull/156
* Remove the `median` of the Bernoulli distribution. by copybara-service in https://github.com/deepmind/distrax/pull/157
* Change tolerance of sample stddev test in `bertnoulli_test.py`. by copybara-service in https://github.com/deepmind/distrax/pull/160
* Delete references to removed jax.core attributes. by copybara-service in https://github.com/deepmind/distrax/pull/166
* Update .pylintrc by copybara-service in https://github.com/deepmind/distrax/pull/170
* Add the von Mises distribution to distrax. by copybara-service in https://github.com/deepmind/distrax/pull/169
* Allow Distribution to work with ArrayTree events. by copybara-service in https://github.com/deepmind/distrax/pull/175
* Explicitly separate JAX and non-JAX data during Jittable serialization. by copybara-service in https://github.com/deepmind/distrax/pull/177
* Add PyPI badge showing the latest released version. by copybara-service in https://github.com/deepmind/distrax/pull/179
* Add `vmap` support for Categorical, Normal, and Independent. by copybara-service in https://github.com/deepmind/distrax/pull/180
* Use jax.tree_util.$tree_fn instead of deprecated jax.$tree_fn alias. by copybara-service in https://github.com/deepmind/distrax/pull/181
* Compare with slightly less numerical precision. by copybara-service in https://github.com/deepmind/distrax/pull/183
* Avoid NaN gradients from Categorical KL and entropy. by copybara-service in https://github.com/deepmind/distrax/pull/190
* Add categorical-uniform distribution. by copybara-service in https://github.com/deepmind/distrax/pull/199
* Add categorical-uniform distribution. by copybara-service in https://github.com/deepmind/distrax/pull/205
* Add clipped distributions to distrax. by copybara-service in https://github.com/deepmind/distrax/pull/210
* Lower the requirements to test the VonMises distribution. by copybara-service in https://github.com/deepmind/distrax/pull/217
* Increase tolerance of test in `gumbel_test.py`. by copybara-service in https://github.com/deepmind/distrax/pull/221
* Prepare inverse transformation for JAX jit==pjit migration. by copybara-service in https://github.com/deepmind/distrax/pull/223
* Fix `TypeError`s for python < 3.10. Fixes 224. by copybara-service in https://github.com/deepmind/distrax/pull/225
**Full Changelog**: https://github.com/deepmind/distrax/compare/v0.1.2...v0.1.3