aspire.flows.jax.utils
======================

.. py:module:: aspire.flows.jax.utils


Functions
---------

.. autoapisummary::

   aspire.flows.jax.utils.get_flow_function_class
   aspire.flows.jax.utils.get_bijection_class
   aspire.flows.jax.utils.get_flow


Module Contents
---------------

.. py:function:: get_flow_function_class(name)

.. py:function:: get_bijection_class(name)

.. py:function:: get_flow(*, key, dims, flow_type = 'masked_autoregressive_flow', bijection_type = None, bijection_kwargs = None, dtype=None, **kwargs)