`nessai.flows.transforms`

Transform for use in normalising flows.

Module Contents

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: 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 https://github.com/bayesiains/nflows/pull/38 for details.

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