``jax.dtypes`` module

docs/jax.dtypes.rst

0.3.25243 B

`jax.dtypes` module

.. automodule:: jax.dtypes

.. autosummary:: :toctree: _autosummary

bfloat16
canonicalize_dtype
float0
itemsize_bits
issubdtype
prng_key
result_type
scalar_type_of