credit.mixed_precision
======================

.. py:module:: credit.mixed_precision


Functions
---------

.. autoapisummary::

   credit.mixed_precision.parse_dtype


Module Contents
---------------

.. py:function:: parse_dtype(dtype_str: str) -> torch.dtype

   Parses a string representation of a torch dtype and returns the corresponding torch.dtype object.

   :param dtype_str: The string representation of the dtype (e.g., "float32", "int64").
   :type dtype_str: str

   :returns: The corresponding torch.dtype object.
   :rtype: torch.dtype