nessai.flows.transforms

nessai.flows.transforms#

Transform for use in normalising flows.

Classes#

LULinear

Wrapper for LULinear from nflows that works with CUDA.

Module Contents#

class nessai.flows.transforms.LULinear(features, using_cache=False, identity_init=True, eps=0.001)#

Bases: glasflow.nflows.transforms.LULinear

Wrapper for LULinear from nflows that works with CUDA.

The original implementation has a bug that prevents use with CUDA. See bayesiains/nflows#38 for details.

This should be removed if the bug is fixed in nflows.