Raw File
jax.dtypes.rst
``jax.dtypes`` module
=====================

.. automodule:: jax.dtypes

.. autosummary::
  :toctree: _autosummary

    bfloat16
    canonicalize_dtype
    float0
    issubdtype
    prng_key
    result_type
    scalar_type_of
back to top