diff --git a/src/python/torchdistx/__init__.py b/src/python/torchdistx/__init__.py
index 42970e6..73f4040 100644
--- a/src/python/torchdistx/__init__.py
+++ b/src/python/torchdistx/__init__.py
@@ -5,3 +5,6 @@
 # LICENSE file in the root directory of this source tree.
 
 __version__ = "0.3.0.dev0"
+
+from torchdistx import optimizers as optimizers
+from torchdistx import slowmo as slowmo