torchgear
=========

**torchgear** provides utility extensions for PyTorch: tensor math helpers,
shape utilities, and :class:`torch.nn.Module` mixins for device/dtype
consistency and parameter freezing.

.. toctree::
   :maxdepth: 2
   :caption: User guide

   getting_started

.. toctree::
   :maxdepth: 2
   :caption: API reference

   api/math
   api/tensor
   api/module

Indices and tables
==================

* :ref:`genindex`
* :ref:`modindex`
* :ref:`search`