v15.0.0

Breaking changes:

• Users must switch from old style PRNGKey arrays to new style ones (replacing jax.random.PRNGKey with jax.random.key . The old keys will be deprecated in JAX.
• LogNormal is now a Exp transformed Normal, which is an implementation detail, unless you previously relied on log_normal.base_dist or log_normal.bijection.
• recursive_unwrap was removed, as alone it didn't provide additional functionality.

What's Changed

• Avoid array in AutoregressiveBisectionInverter by @danielward27 in #174
• Simplify wrappers by @danielward27 in #175
• Docs and switch to jr.key by @danielward27 in #177
• Documentation example fixes by @danielward27 in #178
• Update README.md by @danielward27 in #179

Full Changelog: v14.0.0...v15.0.0

v14.0.0

What's Changed

I was not happy with the original implementation of some wrappers in flowjax.wrappers, e.g. BijectionReparam introduced some (avoidable, but untidy) circular dependency issues. We now primarily use Parameterize a new name for what was Lambda. This introduces some breaking changes, primarily the aformentioned renaming, in addition to the removal of BijectionReparam and Where . See the pull request for more information.

• Update wrappers by @danielward27 in #172

Full Changelog: v13.1.1...v14.0.0

v13.1.1

What's Changed

• Test for updated equinox errors by @danielward27 in #171

Full Changelog: v13.1.0...v13.1.1

v13.1.0

What's Changed

• Planar inverse with leaky relu by @danielward27 in #170.

Full Changelog: v13.0.1...v13.1.0

Thanks for the help @weiyaw!

v13.0.1

New release to trigger update to docs.

What's Changed

• Update docs by @danielward27 in #168

v13.0.0

Some small breaking changes:

• The interval attribute of RationalQuadraticSpline is now a two-tuple with type tuple[float | int, float | int], representing the upper and lower bound of the spline. Previously, the interval was a float | int, forcing the interval to be symmetric about 0.
• Any custom loss functions used with fit_to_data now should accept a key (whether or not it is used), having signature loss(params, static, x, condition, key), for consistency of API.

What's Changed

• Allow uneven interval in spline by @danielward27 in #166
• Sample contrastive by @danielward27 in #167

Full Changelog: v12.4.0...v13.0.0

v12.4.0

What's Changed

• Add non_trainable by @danielward27 in #165

Full Changelog: v12.3.0...v12.4.0

v12.3.0

What's Changed

The main thing of note is a bug fix to ContrastiveLoss (see #164). Apologies for any inconvenience!

• Fix bug in ContrastiveLoss normalization and add improve numpyro transformed wrapper by @danielward27 in #164

Full Changelog: v12.2.0...v12.3.0

v12.2.0

What's Changed

• More consistent parameter type conversion #158
• Add logistic distribution #159
• Int type dummy array in wrappers #160
• Fix bug in masked autoregressive flow masking #162
• Add a mixture distribution #163

Full Changelog: v12.1.2...v12.2.0

v12.1.2

What's Changed

• Avoid doctest for now by @danielward27 in #156
• Update github workflows and version by @danielward27 in #157

Full Changelog: v12.1.1...v12.1.2