Releases: google-deepmind/distrax
Releases · google-deepmind/distrax
v0.1.8
What's Changed
- Migrate from
jax.experimental.enable_x64tojax.enable_x64. by @copybara-service[bot] in #307 - Replace deprecated
jax.core.get_avalwithjax.typeofby @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.abstractifyby @copybara-service[bot] in #289 - remove
pjit_p, leaving onlyjax.extend.core.primitives.jit_pby @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_functionforLaplacedistribution. 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_functionforLaplacedistribution. by @copybara-service in #251
Full Changelog: v0.1.4...v0.1.5
Distrax 0.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_cdfmethod for the Gamma distribution and modified some tests ingamma_test.py. by @copybara-service in #141 - Fix
test_convert_seed. by @copybara-service in #142 - Implement the
Betadistribution. 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.Distributionbase class. by @copybara-service in #148 - Implement survival and log-survival function for the
Logisticdistribution. by @copybara-service in #150 - Implement survival and log-survival function for the
Normaldistribution. by @copybara-service in #151 - Raise
ValueErrorwhen input parameters are not valid forBernoulli,Multinomial, andCategorical. by @copybara-service in #153 - Remove
RTOLfrom 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
medianof 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
vmapsupport 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
matrixproperty 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
DiagAffinebijector. by @copybara-service in #115 - Implement the
MultivariateNormalFromBijectordistribution. by @copybara-service in #116 - Implement the
MultivariateNormalDiagdistribution as an instance of aMultivariateNormalFromBijector. by @copybara-service in #117 - Implement the
MultivariateNormalTridistribution. by @copybara-service in #118 - Implement the
MultivariateNormalDiagPlusLowRankdistribution. by @copybara-service in #119 - Add tests for KL computations between different Multivariate Gaussian distributions. by @copybara-service in #123
- Demote
TriangularAffinetoTriangularLinear. by @copybara-service in #125 - Add tests for
same_asmethod to increase test coverage. by @copybara-service in #126 - Demote
DiagAffinetoDiagLinear. by @copybara-service in #127 - Demote
DiagPlusLowRankAffinetoDiagPlusLowRankLinear. 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