Surjection layers for density estimation with normalizing flows

Surjectors is a light-weight library of inference and generative surjection layers, i.e., layers that reduce or increase dimensionality, for density estimation using normalizing flows. Surjectors builds on Distrax and Haiku and is fully compatible with both of them.

TODO

Make sure to have a working JAX installation. Depending whether you want to use CPU/GPU/TPU, please follow these instructions.

To install the latest GitHub , just call the following on the command line:

pip install git+https://github.com/dirmeier/surjectors@<RELEASE>

Simon Dirmeier sfyrbnd @ pm me