Torch modules
==================
``irtorch.torch_modules`` contains various PyTorch modules. These modules are used in different contexts of the package, but can also be used separately.

.. autoclass:: irtorch.torch_modules.MonotonePolynomialModule
    :show-inheritance:
    :members:

.. autoclass:: irtorch.torch_modules.NeuralSplineFlow
    :show-inheritance:
    :members:

.. autoclass:: irtorch.torch_modules.RationalQuadraticSpline
    :show-inheritance:
    :members: