Home
INNLab is an invertible neural network (INN) library for easier implementation of INNs. Using INNLab, you can easily build INNs and compose multiple types of INNs together for complex tasks.
INNLab has similar structure and format as PyTorch, which makes it easier to getting start with.
- INN layers
- Kernel layers
- Containers
- Distributions