Merge pull request #79 from KristianHolsheimer:master

PiperOrigin-RevId: 365218050
google-deepmind · Mar 26, 2021 · 0bd9c65 · 0bd9c65
2 parents 4a9825b + aa374ab
commit 0bd9c65
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/docs/api.rst b/docs/api.rst
@@ -251,7 +251,7 @@ multi_normal
 .. autofunction:: multi_normal
 
 Apply Updates
--------------
+=============
 
 .. autosummary::
     apply_updates
@@ -406,7 +406,7 @@ losses
 
 
 Linear Algebra Operators
--------------------------
+========================
 
 .. currentmodule:: optax
 
@@ -500,7 +500,7 @@ hvp
 
 
 Control Variates
-----------------
+================
 
 .. currentmodule:: optax
 
@@ -529,7 +529,7 @@ moving_avg_baseline
 
 
 Stochastic Gradient Estimators
-------------------------------
+==============================
 
 .. currentmodule:: optax