flowjax 0.2.1

Basic flowjax implementation in jax.

flowjax

---

Normalising flow implementations in jax.

So far supports:

• Affine coupling flows/RealNVP (Dinh et al.) - https://arxiv.org/abs/1605.08803

• Neural spline coupling flows (Durkan et al.) - https://arxiv.org/abs/1906.04032/

See examples at https://github.com/danielward27/flowjax/blob/main/examples/

Authors

---

flowjax was written by Daniel Ward <[e-mail]>_.