diff --git a/optax/contrib/__init__.py b/optax/contrib/__init__.py
index be8be8404..c9db9617a 100644
--- a/optax/contrib/__init__.py
+++ b/optax/contrib/__init__.py
@@ -14,5 +14,5 @@
 # ==============================================================================
 """Contributed optimizers in Optax."""
 
-from optax._src.contrib.mechanic import MechanicState
-from optax._src.contrib.mechanic import mechanize
+from optax.contrib.mechanic import MechanicState
+from optax.contrib.mechanic import mechanize
diff --git a/optax/_src/contrib/mechanic.py b/optax/contrib/mechanic.py
similarity index 100%
rename from optax/_src/contrib/mechanic.py
rename to optax/contrib/mechanic.py
diff --git a/optax/_src/contrib/mechanic_test.py b/optax/contrib/mechanic_test.py
similarity index 99%
rename from optax/_src/contrib/mechanic_test.py
rename to optax/contrib/mechanic_test.py
index f13f2aee6..cc70903bc 100644
--- a/optax/_src/contrib/mechanic_test.py
+++ b/optax/contrib/mechanic_test.py
@@ -27,7 +27,7 @@
 from optax._src import numerics
 from optax._src import state_utils
 from optax._src import update
-from optax._src.contrib import mechanic
+from optax.contrib import mechanic
 
 
 # TODO(harshm): make LARS and Fromage work with mechanic.