.gitignore
LICENSE
README.md
logo.png
pyproject.toml
.github/workflows/publish.yml
.github/workflows/test.yml
docs/Makefile
docs/README.md
docs/conf.py
docs/faq.rst
docs/index.rst
docs/make.bat
docs/_static/style.css
docs/api/bijections.rst
docs/api/distributions.rst
docs/api/experimental.rst
docs/api/flows.rst
docs/api/losses.rst
docs/api/training.rst
docs/api/wrappers.rst
docs/examples/bounded.ipynb
docs/examples/conditional.ipynb
docs/examples/snpe.ipynb
docs/examples/unconditional.ipynb
docs/examples/variational_inference.ipynb
flowjax/__init__.py
flowjax/bisection_search.py
flowjax/distributions.py
flowjax/flows.py
flowjax/masks.py
flowjax/tasks.py
flowjax/utils.py
flowjax/wrappers.py
flowjax.egg-info/PKG-INFO
flowjax.egg-info/SOURCES.txt
flowjax.egg-info/dependency_links.txt
flowjax.egg-info/requires.txt
flowjax.egg-info/top_level.txt
flowjax/bijections/__init__.py
flowjax/bijections/affine.py
flowjax/bijections/bijection.py
flowjax/bijections/block_autoregressive_network.py
flowjax/bijections/chain.py
flowjax/bijections/concatenate.py
flowjax/bijections/coupling.py
flowjax/bijections/exp.py
flowjax/bijections/jax_transforms.py
flowjax/bijections/masked_autoregressive.py
flowjax/bijections/planar.py
flowjax/bijections/rational_quadratic_spline.py
flowjax/bijections/softplus.py
flowjax/bijections/tanh.py
flowjax/bijections/utils.py
flowjax/experimental/__init__.py
flowjax/experimental/numpyro.py
flowjax/train/__init__.py
flowjax/train/data_fit.py
flowjax/train/losses.py
flowjax/train/train_utils.py
flowjax/train/variational_fit.py
tests/test_bisection_search.py
tests/test_distributions.py
tests/test_flows.py
tests/test_masks.py
tests/test_utils.py
tests/test_wrappers.py
tests/test_bijections/test_bijection_utils.py
tests/test_bijections/test_bijections.py
tests/test_bijections/test_bnaf.py
tests/test_bijections/test_chain.py
tests/test_bijections/test_jax_transforms.py
tests/test_bijections/test_masked_autoregressive.py
tests/test_bijections/test_rational_quadratic_spline.py
tests/test_experimental/test_numpyro.py
tests/test_train/test_data_fit.py
tests/test_train/test_train_utils.py
tests/test_train/test_variational_fit.py