diff --git a/experiments/mnist/flax/README.md b/experiments/mnist/flax_example/README.md
similarity index 100%
rename from experiments/mnist/flax/README.md
rename to experiments/mnist/flax_example/README.md
diff --git a/experiments/mnist/flax/configs/__init__.py b/experiments/mnist/flax_example/configs/__init__.py
similarity index 100%
rename from experiments/mnist/flax/configs/__init__.py
rename to experiments/mnist/flax_example/configs/__init__.py
diff --git a/experiments/mnist/flax/configs/default.py b/experiments/mnist/flax_example/configs/default.py
similarity index 100%
rename from experiments/mnist/flax/configs/default.py
rename to experiments/mnist/flax_example/configs/default.py
diff --git a/experiments/mnist/flax/main.py b/experiments/mnist/flax_example/main.py
similarity index 100%
rename from experiments/mnist/flax/main.py
rename to experiments/mnist/flax_example/main.py
diff --git a/experiments/mnist/flax/mnist_benchmark.py b/experiments/mnist/flax_example/mnist_benchmark.py
similarity index 100%
rename from experiments/mnist/flax/mnist_benchmark.py
rename to experiments/mnist/flax_example/mnist_benchmark.py
diff --git a/experiments/mnist/flax/requirements.txt b/experiments/mnist/flax_example/requirements.txt
similarity index 100%
rename from experiments/mnist/flax/requirements.txt
rename to experiments/mnist/flax_example/requirements.txt
diff --git a/experiments/mnist/flax/train.py b/experiments/mnist/flax_example/train.py
similarity index 100%
rename from experiments/mnist/flax/train.py
rename to experiments/mnist/flax_example/train.py