.gitattributes
.gitignore
.gitlint
.pre-commit-config.yaml
.python-version
LICENSE
README.md
pyproject.toml
uv.lock
.github/workflows/ci.yaml
.github/workflows/release.yaml
examples/density_estimation.md
examples/dimensionality_reduction.md
examples/introduction.md
experiments/spike/benchmarking-mlx/.gitignore
experiments/spike/benchmarking-mlx/.python-version
experiments/spike/benchmarking-mlx/README.md
experiments/spike/benchmarking-mlx/data.py
experiments/spike/benchmarking-mlx/flow_jax.py
experiments/spike/benchmarking-mlx/flow_mlx.py
experiments/spike/benchmarking-mlx/main.py
experiments/spike/benchmarking-mlx/pyproject.toml
figures/introduction_00.png
figures/introduction_01.png
figures/normalizing_flows_00.png
figures/normalizing_flows_01.png
figures/normalizing_flows_02.png
figures/normalizing_flows_03.png
paper/paper.bib
paper/paper.md
surjectors/__init__.py
surjectors/util.py
surjectors/util_test.py
surjectors.egg-info/PKG-INFO
surjectors.egg-info/SOURCES.txt
surjectors.egg-info/dependency_links.txt
surjectors.egg-info/requires.txt
surjectors.egg-info/top_level.txt
surjectors/_src/__init__.py
surjectors/_src/_math.py
surjectors/_src/_transform.py
surjectors/_src/bijectors/__init__.py
surjectors/_src/bijectors/affine_masked_autoregressive.py
surjectors/_src/bijectors/affine_masked_autoregressive_test.py
surjectors/_src/bijectors/affine_masked_coupling.py
surjectors/_src/bijectors/affine_masked_coupling_test.py
surjectors/_src/bijectors/bijector.py
surjectors/_src/bijectors/lu_linear.py
surjectors/_src/bijectors/lu_linear_math_test.py
surjectors/_src/bijectors/lu_liner_test.py
surjectors/_src/bijectors/masked_autoregressive.py
surjectors/_src/bijectors/masked_autoregressive_math_test.py
surjectors/_src/bijectors/masked_autoregressive_test.py
surjectors/_src/bijectors/masked_coupling.py
surjectors/_src/bijectors/masked_coupling_math_test.py
surjectors/_src/bijectors/masked_coupling_test.py
surjectors/_src/bijectors/permutation.py
surjectors/_src/bijectors/rq_masked_coupling.py
surjectors/_src/bijectors/rq_masked_coupling_test.py
surjectors/_src/bijectors/wrap.py
surjectors/_src/bijectors/wrap_test.py
surjectors/_src/conditioners/__init__.py
surjectors/_src/conditioners/mlp.py
surjectors/_src/conditioners/transformer.py
surjectors/_src/conditioners/nn/__init__.py
surjectors/_src/conditioners/nn/made.py
surjectors/_src/conditioners/nn/made_test.py
surjectors/_src/conditioners/nn/masked_linear.py
surjectors/_src/distributions/__init__.py
surjectors/_src/distributions/transformed_distribution.py
surjectors/_src/surjectors/__init__.py
surjectors/_src/surjectors/affine_masked_autoregressive_inference_funnel.py
surjectors/_src/surjectors/affine_masked_autoregressive_inference_funnel_test.py
surjectors/_src/surjectors/affine_masked_coupling_generative_funnel.py
surjectors/_src/surjectors/affine_masked_coupling_inference_funnel.py
surjectors/_src/surjectors/affine_masked_coupling_inference_funnel_test.py
surjectors/_src/surjectors/augment.py
surjectors/_src/surjectors/augment_math_test.py
surjectors/_src/surjectors/chain.py
surjectors/_src/surjectors/funnel.py
surjectors/_src/surjectors/funnel_test.py
surjectors/_src/surjectors/masked_autoregressive_inference_funnel.py
surjectors/_src/surjectors/masked_autoregressive_inference_funnel_math_test.py
surjectors/_src/surjectors/masked_coupling_inference_funnel.py
surjectors/_src/surjectors/masked_coupling_inference_funnel_math_test.py
surjectors/_src/surjectors/mlp.py
surjectors/_src/surjectors/mlp_math_test.py
surjectors/_src/surjectors/rq_masked_autoregressive_inference_funnel.py
surjectors/_src/surjectors/rq_masked_coupling_inference_funnel.py
surjectors/_src/surjectors/rq_masked_coupling_inference_funnel_test.py
surjectors/_src/surjectors/slice.py
surjectors/_src/surjectors/slice_math_test.py
surjectors/_src/surjectors/slice_test.py
surjectors/_src/surjectors/surjector.py
surjectors/_src/surjectors/surjectors_test.py
surjectors/_src/testing/__init__.py
surjectors/_src/testing/layer_harness.py
surjectors/_src/testing/layer_harness_test.py
surjectors/nn/__init__.py