v0.1.8

What's Changed

• Migrate from jax.experimental.enable_x64 to jax.enable_x64. by @copybara-service[bot] in #307
• Replace deprecated jax.core.get_aval with jax.typeof by @copybara-service[bot] in #319
• [JAX] Remove stale references to //jax:experimental, which is being removed. by @copybara-service[bot] in #322
• Fix Distrax from Jax API change. by @copybara-service[bot] in #325

Full Changelog: v0.1.7...v0.1.8

Distrax 0.1.7

What's Changed

• migrate to pyproject.toml by @copybara-service[bot] in #303
• release v0.1.7 by @copybara-service[bot] in #304

Full Changelog: v0.1.6...v0.1.7

Distrax 0.1.6

What's Changed

• The main tensorflow_probability BUILD target is being changed to only load the by @copybara-service[bot] in #272
• Add sqrt/square primitives to the inverse registry after openxla/stablehlo#2623 by @copybara-service[bot] in #279
• Remove references to jax.core.raise_to_shaped by @copybara-service[bot] in #281
• Add accuracy field to unary ops by @copybara-service[bot] in #285
• Configure tests to explicitly use jax_threefry_partitionable=False. by @copybara-service[bot] in #286
• Configure tests to explicitly use jax_threefry_partitionable=False. by @copybara-service[bot] in #288
• [distrax] Replace usages of deprecated jax.interpreters.xla.abstractify by @copybara-service[bot] in #289
• remove pjit_p, leaving only jax.extend.core.primitives.jit_p by @copybara-service[bot] in #293
• Disable bad-ident warnings. by @copybara-service[bot] in #300
• release v0.1.6 by @copybara-service[bot] in #302

Full Changelog: v0.1.42...v0.1.6

Distrax 0.1.43

v0.1.43

Install dependencies before running version checks in the release wor…

Distrax 0.1.42

v0.1.42

Install dependencies before running version checks in the release wor…

Distrax 0.1.41

What's Changed

• Remove references to deprecated jax aliases by @copybara-service in #243
• Add log_survival_function for Laplace distribution. by @copybara-service in #251

Full Changelog: v0.1.4...v0.1.

Distrax 0.1.5

What's Changed

• Remove references to deprecated jax aliases by @copybara-service in #243
• Add log_survival_function for Laplace distribution. by @copybara-service in #251

Full Changelog: v0.1.4...v0.1.5

Distrax 0.1.4

What's Changed

• Update categorical.py by @cyprienc in #236
• Drop python 3.7 and 3.8

New Contributors

• @cyprienc made their first contribution in #236

Full Changelog: v0.1.3...v0.1.4

Version v0.1.3

What's Changed

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

Full Changelog: v0.1.2...v0.1.3

Distrax 0.1.2

What's Changed

• Implement TFP's base measure interface for distrax distributions and bijectors. by @copybara-service in #104
• Add a matrix property to affine bijectors that returns a matrix representing their linear part. by @copybara-service in #114
• Fix categorical KL computation when the second distribution has zero probability on entries where the first distribution also has zero probability. by @copybara-service in #113
• Implement the DiagAffine bijector. by @copybara-service in #115
• Implement the MultivariateNormalFromBijector distribution. by @copybara-service in #116
• Implement the MultivariateNormalDiag distribution as an instance of a MultivariateNormalFromBijector. by @copybara-service in #117
• Implement the MultivariateNormalTri distribution. by @copybara-service in #118
• Implement the MultivariateNormalDiagPlusLowRank distribution. by @copybara-service in #119
• Add tests for KL computations between different Multivariate Gaussian distributions. by @copybara-service in #123
• Demote TriangularAffine to TriangularLinear. by @copybara-service in #125
• Add tests for same_as method to increase test coverage. by @copybara-service in #126
• Demote DiagAffine to DiagLinear. by @copybara-service in #127
• Demote DiagPlusLowRankAffine to DiagPlusLowRankLinear. by @copybara-service in #128
• Updated 2-distributions tests for the MultivariateNormalDiagPlusLowRank. by @copybara-service in #131
• Adjust test tolerances. by @copybara-service in #133
• Expose the full-covariance multivariate normal distributions, and upgrade version. by @copybara-service in #134

Full Changelog: v0.1.1...v0.1.2