diff --git a/src/NanoParticleTools/machine_learning/__init__.py b/src/NanoParticleTools/machine_learning/__init__.py
new file mode 100644
index 0000000..d983ab1
--- /dev/null
+++ b/src/NanoParticleTools/machine_learning/__init__.py
@@ -0,0 +1,3 @@
+from NanoParticleTools.machine_learning.data.dataset import NPMCDataset
+from NanoParticleTools.machine_learning.data.processors import SummedWavelengthRangeLabelProcessor
+from NanoParticleTools.machine_learning.data.datamodule import NPMCDataModule
diff --git a/src/NanoParticleTools/optimization/__init__.py b/src/NanoParticleTools/optimization/__init__.py
new file mode 100644
index 0000000..e69de29