Neural Network Module API Reference
===================================

.. automodule:: scistanpy.model.nn_module
   :undoc-members:
   :show-inheritance:

Core PyTorch Integration
------------------------

.. autoclass:: scistanpy.model.nn_module.PyTorchModel
   :members:
   :undoc-members:
   :show-inheritance: