Surjection layers for density estimation with normalizing flows
Project description
surjectors
Surjection layers for density estimation with normalizing flows
About
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.
Examples
You can find several self-contained examples on how to use the algorithms in examples.
Installation
Make sure to have a working JAX installation. Depending whether you want to use CPU/GPU/TPU,
please follow these instructions.
To install the package from PyPI, call:
pip install surjectors
To install the latest GitHub , just call the following on the command line:
pip install git+https://github.com/dirmeier/surjectors@<RELEASE>
Author
Simon Dirmeier sfyrbnd @ pm me
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for surjectors-0.2.4-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | dbbd0e68a0c4fd2f4d60e077b5823be597c38a8b582b874f91433ab612c9b1e5 |
|
| MD5 | e64afe1814442bfe4045db91fa6718e5 |
|
| BLAKE2b-256 | 72211266df858aa0e5d4d78a5c6741ce787efdaf8de463493870a7a0e799664b |
