Metadata-Version: 2.1
Name: flowjax
Version: 0.2.1
Summary: Basic flowjax implementation in jax.
Home-page: https://github.com/danielward27/flowjax.git
Author: Daniel Ward
Author-email: danielward27@outlook.com
License: MIT
Platform: UNKNOWN
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: jax
Requires-Dist: equinox
Requires-Dist: pytest
Requires-Dist: tqdm
Requires-Dist: optax

## 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 <danielward27@outlook.com>`_.


