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

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


Submodules
----------

.. toctree::
   :maxdepth: 1

   /autoapi/aspire/flows/jax/flows/index
   /autoapi/aspire/flows/jax/utils/index


Functions
---------

.. autoapisummary::

   aspire.flows.jax.get_flow


Package Contents
----------------

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