diff --git a/docs/source/trax.supervised.rst b/docs/source/trax.supervised.rst
index 173ac5d55..de2b652de 100644
--- a/docs/source/trax.supervised.rst
+++ b/docs/source/trax.supervised.rst
@@ -5,13 +5,16 @@ decoding
 --------
 
 .. automodule:: trax.supervised.decoding
+   :members:
 
 lr\_schedules
 -------------
 
 .. automodule:: trax.supervised.lr_schedules
+   :members:
 
 training
 --------
 
 .. automodule:: trax.supervised.training
+   :members: