:py:mod:`nessai.flows.transforms` ================================= .. py:module:: nessai.flows.transforms .. autoapi-nested-parse:: Transform for use in normalising flows. .. !! processed by numpydoc !! Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: nessai.flows.transforms.LULinear .. py:class:: LULinear(features, using_cache=False, identity_init=True, eps=0.001) Bases: :py:obj:`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. .. !! processed by numpydoc !! .. py:method:: weight_inverse(self) Cost: inverse = O(D^3) where: D = num of features .. !! processed by numpydoc !!