mixed_precision
=========================

.. py:module:: jaxns.internals.mixed_precision

.. rubric:: :code:`jaxns.internals.mixed_precision`


.. rubric:: Module Contents

.. py:data:: T

.. py:data:: X

.. py:class:: Policy

   Encapsulates casting for inputs, outputs and parameters.


   .. py:attribute:: measure_dtype
      :type:  jax.numpy.dtype


   .. py:attribute:: index_dtype
      :type:  jax.numpy.dtype


   .. py:attribute:: count_dtype
      :type:  jax.numpy.dtype


   .. py:method:: cast_to_index(x, quiet = False)

      Converts index values to the index dtype.



   .. py:method:: cast_to_measure(x, quiet = False)

      Converts measure values to the measure dtype.



   .. py:method:: cast_to_count(x, quiet = False)

      Converts count values to the count dtype.



.. py:data:: mp_policy

.. py:data:: float_type

.. py:data:: int_type

.. py:data:: complex_type