diff --git a/.github/workflows/update_hpc_requirements.yml b/.github/workflows/update_hpc_requirements.yml
index 7e09b14..38dbc42 100644
--- a/.github/workflows/update_hpc_requirements.yml
+++ b/.github/workflows/update_hpc_requirements.yml
@@ -28,7 +28,7 @@ jobs:
         uses: astral-sh/setup-uv@v5
 
       - name: Regenerate env/hpc/requirements.txt
-        run: uv run scripts/export_hpc_requirements.py
+        run: uv run scripts/hpc/export_requirements.py
 
       - name: Commit updated requirements if changed
         uses: stefanzweifel/git-auto-commit-action@v5
diff --git a/docs/HPC.md b/docs/HPC.md
index ff18a5f..3e0e8d1 100644
--- a/docs/HPC.md
+++ b/docs/HPC.md
@@ -79,7 +79,7 @@ After installation, run these commands to ensure your environment is set up corr
 `env/hpc/requirements.txt` is auto-generated from `pyproject.toml`. To regenerate:
 
 ```bash
-uv run scripts/export_hpc_requirements.py
+uv run scripts/hpc/export_requirements.py
 ```
 
 Modules listed in `env/hpc/modules.txt` are automatically excluded from the pip requirements to save space and use HPC-optimized binaries.
diff --git a/docs/api/train.md b/docs/api/train.md
deleted file mode 100644
index f87f5c1..0000000
--- a/docs/api/train.md
+++ /dev/null
@@ -1 +0,0 @@
-# TODO
\ No newline at end of file
diff --git a/experiments/plots/__init__.py b/experiments/plots/__init__.py
deleted file mode 100644
index 92f34cb..0000000
--- a/experiments/plots/__init__.py
+++ /dev/null
@@ -1,3 +0,0 @@
-from .plot import simple_plot
-
-__all__ = ["simple_plot"]
diff --git a/experiments/plots/plot.py b/experiments/plots/plot.py
deleted file mode 100644
index ab8f4ad..0000000
--- a/experiments/plots/plot.py
+++ /dev/null
@@ -1,12 +0,0 @@
-import matplotlib.pyplot as plt
-
-
-def simple_plot(x: list, y: list, show_window: bool = False, filename: str = "plot.png") -> None:
-    plt.plot(x, y)
-    plt.savefig(filename)
-
-    if show_window:
-        # blocks until window is closed
-        plt.show()
-
-    plt.close()
diff --git a/scripts/export_hpc_requirements.py b/scripts/hpc/export_requirements.py
similarity index 98%
rename from scripts/export_hpc_requirements.py
rename to scripts/hpc/export_requirements.py
index 9ff3de7..5908ad3 100644
--- a/scripts/export_hpc_requirements.py
+++ b/scripts/hpc/export_requirements.py
@@ -5,9 +5,6 @@ This is a LOCAL DEVELOPER UTILITY — run it on your own machine before pushing
 code whenever pyproject.toml dependencies change. It reads the modules from
 env/hpc/modules.txt and the full dependency list from pyproject.toml, then
 writes the remainder to env/hpc/requirements.txt.
-
-Usage:
-    uv run scripts/export_hpc_requirements.py
 """
 
 from __future__ import annotations
diff --git a/experiments/simulate.py b/scripts/simulate.py
similarity index 100%
rename from experiments/simulate.py
rename to scripts/simulate.py
diff --git a/experiments/train.py b/scripts/train.py
similarity index 97%
rename from experiments/train.py
rename to scripts/train.py
index 0a65493..9ee6da6 100644
--- a/experiments/train.py
+++ b/scripts/train.py
@@ -7,7 +7,7 @@ import yaml
 import os
 
 from brittle_star_project.dataclasses import PPOArgs
-from PPOTrainer import PPOTrainer
+from brittle_star_project.trainers.PPOTrainer import PPOTrainer
 from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleStarJaxEnvWrapper
 
 
diff --git a/experiments/PPOTrainer.py b/src/brittle_star_project/trainers/PPOTrainer.py
similarity index 98%
rename from experiments/PPOTrainer.py
rename to src/brittle_star_project/trainers/PPOTrainer.py
index 365191c..7df57f0 100644
--- a/experiments/PPOTrainer.py
+++ b/src/brittle_star_project/trainers/PPOTrainer.py
@@ -28,13 +28,13 @@ from ppo import PPO
 
 
 @jax.jit
-def linear_schedule(count, minibatch_count, update_epochs, num_iterations, learning_rate):
+def _linear_schedule(count, minibatch_count, update_epochs, num_iterations, learning_rate):
     frac = 1.0 - (count // (minibatch_count * update_epochs)) / num_iterations
     return learning_rate * frac
 
 
 @jax.jit
-def convert_obs_dict_to_array(obs_dict: dict) -> jnp.ndarray:
+def _convert_obs_dict_to_array(obs_dict: dict) -> jnp.ndarray:
     return jax.vmap(lambda o: jnp.concatenate([v.flatten() for v in o.values() if v.size > 0]))(
         obs_dict
     )
@@ -124,7 +124,7 @@ def _step_env_wrapped(episode_stats, env_state, action, env_step_fn):
     return (
         episode_stats,
         next_env_state,
-        (convert_obs_dict_to_array(next_env_state.observations), reward, done),
+        (_convert_obs_dict_to_array(next_env_state.observations), reward, done),
     )
 
 
@@ -300,7 +300,7 @@ class PPOTrainer:
                 optax.clip_by_global_norm(self.args.max_grad_norm),
                 optax.inject_hyperparams(optax.adam)(
                     learning_rate=partial(
-                        linear_schedule,
+                        _linear_schedule,
                         minibatch_count=self.args.num_minibatches,
                         update_epochs=self.args.update_epochs,
                         num_iterations=self.args.num_iterations,
@@ -467,7 +467,7 @@ class PPOTrainer:
                 print(f">>> [HPC] Initial reset started: {time.ctime()}", flush=True)
 
         env_state = self.env.reset(seed=self.args.seed)
-        next_obs = convert_obs_dict_to_array(env_state.observations)
+        next_obs = _convert_obs_dict_to_array(env_state.observations)
         next_done = jnp.zeros(self.args.num_envs, dtype=jnp.bool_)
 
         if log and not is_tty:
diff --git a/src/brittle_star_project/trainers/__init__.py b/src/brittle_star_project/trainers/__init__.py
new file mode 100644
index 0000000..e69de29