diff --git a/prototorch/__init__.py b/prototorch/__init__.py
index 4953edb..7ba09a0 100644
--- a/prototorch/__init__.py
+++ b/prototorch/__init__.py
@@ -2,9 +2,10 @@
 
 __version__ = '0.1.1-dev0'
 
-from prototorch import functions, modules
+from prototorch import datasets, functions, modules
 
 __all__ = [
+    'datasets',
     'functions',
     'modules',
 ]