*************
API Reference
*************

.. currentmodule:: torch_deterministic

.. autosummary::
   :toctree: api
   :recursive:

   InfiniteSampler
   BatchGenerator
   collate_rngs