trax.fastmath
=============

backend
-------

.. automodule:: trax.fastmath.backend

jax
---

.. automodule:: trax.fastmath.jax

numpy
-----

.. automodule:: trax.fastmath.numpy

tf
--

.. automodule:: trax.fastmath.tf

trax.fastmath
=============

backend
-------

.. automodule:: trax.fastmath.backend

Remove some non-user API from the generated docs.
Remove some non-user API from the generated docs.

PiperOrigin-RevId: 318942376
