diff --git a/src/delphi/__init__.py b/delphi/__init__.py
similarity index 100%
rename from src/delphi/__init__.py
rename to delphi/__init__.py
diff --git a/src/delphi/eval.py b/delphi/eval.py
similarity index 100%
rename from src/delphi/eval.py
rename to delphi/eval.py
diff --git a/src/delphi/test_configs/__init__.py b/delphi/test_configs/__init__.py
similarity index 100%
rename from src/delphi/test_configs/__init__.py
rename to delphi/test_configs/__init__.py
diff --git a/src/delphi/test_configs/debug.json b/delphi/test_configs/debug.json
similarity index 100%
rename from src/delphi/test_configs/debug.json
rename to delphi/test_configs/debug.json
diff --git a/src/delphi/test_configs/debug_transformers_bloom.json b/delphi/test_configs/debug_transformers_bloom.json
similarity index 100%
rename from src/delphi/test_configs/debug_transformers_bloom.json
rename to delphi/test_configs/debug_transformers_bloom.json
diff --git a/src/delphi/test_configs/v0-llama2-100k.json b/delphi/test_configs/v0-llama2-100k.json
similarity index 100%
rename from src/delphi/test_configs/v0-llama2-100k.json
rename to delphi/test_configs/v0-llama2-100k.json
diff --git a/src/delphi/tokenization.py b/delphi/tokenization.py
similarity index 100%
rename from src/delphi/tokenization.py
rename to delphi/tokenization.py
diff --git a/src/delphi/train/__init__.py b/delphi/train/__init__.py
similarity index 100%
rename from src/delphi/train/__init__.py
rename to delphi/train/__init__.py
diff --git a/src/delphi/train/checkpoint_step.py b/delphi/train/checkpoint_step.py
similarity index 100%
rename from src/delphi/train/checkpoint_step.py
rename to delphi/train/checkpoint_step.py
diff --git a/src/delphi/train/config/__init__.py b/delphi/train/config/__init__.py
similarity index 100%
rename from src/delphi/train/config/__init__.py
rename to delphi/train/config/__init__.py
diff --git a/src/delphi/train/config/adam_config.py b/delphi/train/config/adam_config.py
similarity index 100%
rename from src/delphi/train/config/adam_config.py
rename to delphi/train/config/adam_config.py
diff --git a/src/delphi/train/config/dataset_config.py b/delphi/train/config/dataset_config.py
similarity index 100%
rename from src/delphi/train/config/dataset_config.py
rename to delphi/train/config/dataset_config.py
diff --git a/src/delphi/train/config/debug_config.py b/delphi/train/config/debug_config.py
similarity index 100%
rename from src/delphi/train/config/debug_config.py
rename to delphi/train/config/debug_config.py
diff --git a/src/delphi/train/config/training_config.py b/delphi/train/config/training_config.py
similarity index 100%
rename from src/delphi/train/config/training_config.py
rename to delphi/train/config/training_config.py
diff --git a/src/delphi/train/config/utils.py b/delphi/train/config/utils.py
similarity index 100%
rename from src/delphi/train/config/utils.py
rename to delphi/train/config/utils.py
diff --git a/src/delphi/train/config/wandb_config.py b/delphi/train/config/wandb_config.py
similarity index 100%
rename from src/delphi/train/config/wandb_config.py
rename to delphi/train/config/wandb_config.py
diff --git a/src/delphi/train/run_context.py b/delphi/train/run_context.py
similarity index 100%
rename from src/delphi/train/run_context.py
rename to delphi/train/run_context.py
diff --git a/src/delphi/train/shuffle.py b/delphi/train/shuffle.py
similarity index 100%
rename from src/delphi/train/shuffle.py
rename to delphi/train/shuffle.py
diff --git a/src/delphi/train/train_step.py b/delphi/train/train_step.py
similarity index 100%
rename from src/delphi/train/train_step.py
rename to delphi/train/train_step.py
diff --git a/src/delphi/train/training.py b/delphi/train/training.py
similarity index 100%
rename from src/delphi/train/training.py
rename to delphi/train/training.py
diff --git a/src/delphi/train/utils.py b/delphi/train/utils.py
similarity index 100%
rename from src/delphi/train/utils.py
rename to delphi/train/utils.py
diff --git a/src/delphi/train/wandb_utils.py b/delphi/train/wandb_utils.py
similarity index 100%
rename from src/delphi/train/wandb_utils.py
rename to delphi/train/wandb_utils.py
diff --git a/src/delphi/utils.py b/delphi/utils.py
similarity index 100%
rename from src/delphi/utils.py
rename to delphi/utils.py
diff --git a/setup.py b/setup.py
index 4a92f04d..5dd7948a 100644
--- a/setup.py
+++ b/setup.py
@@ -2,8 +2,8 @@
 
 setup(
     name="delphi",
-    packages=find_packages(where="src"),
-    package_dir={"": "src"},
+    packages=find_packages(where="."),
+    package_dir={"": "."},
     package_data={
         "delphi": ["test_configs/**/*"],
     },