nessai.flows.transforms
Transform for use in normalising flows.
Module Contents
Classes
Wrapper for LULinear from nflows that works with CUDA. |
- class nessai.flows.transforms.LULinear(features, using_cache=False, identity_init=True, eps=0.001)
Bases:
nflows.transforms.LULinear
Wrapper for LULinear from nflows that works with CUDA.
The original implementation has a bug that prevents use with CUDA. See https://github.com/bayesiains/nflows/pull/38 for details.
This should be removed if the bug is fixed in nflows.
- weight_inverse(self)
- Cost:
inverse = O(D^3)
- where:
D = num of features