trax.models
===========

atari\_cnn
----------

.. automodule:: trax.models.atari_cnn

beam\_search
------------

.. automodule:: trax.models.beam_search

mlp
---

.. automodule:: trax.models.mlp

neural\_gpu
-----------

.. automodule:: trax.models.neural_gpu

resnet
------

.. automodule:: trax.models.resnet

rl
--

.. automodule:: trax.models.rl

rnn
---

.. automodule:: trax.models.rnn

transformer
-----------

.. automodule:: trax.models.transformer

reformer.reformer
-----------------

.. automodule:: trax.models.reformer.reformer

research.bert
-------------

.. automodule:: trax.models.research.bert

research.skipping\_transformer
------------------------------

.. automodule:: trax.models.research.skipping_transformer

trax.models
===========

atari\_cnn
----------

.. automodule:: trax.models.atari_cnn

mlp
---

.. automodule:: trax.models.mlp

neural\_gpu
-----------

.. automodule:: trax.models.neural_gpu

resnet
------

.. automodule:: trax.models.resnet

rl
--

.. automodule:: trax.models.rl

rnn
---

.. automodule:: trax.models.rnn

transformer
-----------

.. automodule:: trax.models.transformer

reformer.reformer
-----------------

.. automodule:: trax.models.reformer.reformer

research.bert
-------------

.. automodule:: trax.models.research.bert

research.skipping\_transformer
------------------------------

.. automodule:: trax.models.research.skipping_transformer

Remove autodoc reference to non-existing module.
Remove autodoc reference to non-existing module.

PiperOrigin-RevId: 319052787
