diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 165b4a23..99e8cdeb 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -42,7 +42,7 @@ jobs:
         pip install coveralls
     - name: Test with pytest
       run: |
-        coverage run --source=e3nn_jax -m pytest --doctest-modules --ignore=docs/ --ignore=noxfile.py e3nn_jax examples -x
+        coverage run --source=e3nn_jax -m pytest --doctest-modules --ignore=docs/ --ignore=tests/noxfile.py tests examples
     - name: Upload to coveralls
       if: github.event_name == 'push'
       run: |
diff --git a/pyproject.toml b/pyproject.toml
index 14d6e736..3149258d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -89,3 +89,6 @@ exclude_lines = [
 ignore = ["E741", "E203", "W503", "E731", "E721"]
 max-line-length = 127
 max-complexity = 64
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
diff --git a/e3nn_jax/_src/activation_test.py b/tests/_src/activation_test.py
similarity index 100%
rename from e3nn_jax/_src/activation_test.py
rename to tests/_src/activation_test.py
diff --git a/e3nn_jax/_src/basic_test.py b/tests/_src/basic_test.py
similarity index 100%
rename from e3nn_jax/_src/basic_test.py
rename to tests/_src/basic_test.py
diff --git a/e3nn_jax/_src/batchnorm/bn_flax_test.py b/tests/_src/batchnorm/bn_flax_test.py
similarity index 100%
rename from e3nn_jax/_src/batchnorm/bn_flax_test.py
rename to tests/_src/batchnorm/bn_flax_test.py
diff --git a/e3nn_jax/_src/batchnorm/bn_haiku_test.py b/tests/_src/batchnorm/bn_haiku_test.py
similarity index 100%
rename from e3nn_jax/_src/batchnorm/bn_haiku_test.py
rename to tests/_src/batchnorm/bn_haiku_test.py
diff --git a/e3nn_jax/_src/config_test.py b/tests/_src/config_test.py
similarity index 100%
rename from e3nn_jax/_src/config_test.py
rename to tests/_src/config_test.py
diff --git a/e3nn_jax/_src/dropout_haiku_test.py b/tests/_src/dropout_haiku_test.py
similarity index 100%
rename from e3nn_jax/_src/dropout_haiku_test.py
rename to tests/_src/dropout_haiku_test.py
diff --git a/e3nn_jax/_src/gate_test.py b/tests/_src/gate_test.py
similarity index 100%
rename from e3nn_jax/_src/gate_test.py
rename to tests/_src/gate_test.py
diff --git a/e3nn_jax/_src/grad_test.py b/tests/_src/grad_test.py
similarity index 100%
rename from e3nn_jax/_src/grad_test.py
rename to tests/_src/grad_test.py
diff --git a/e3nn_jax/_src/irreps_array_test.py b/tests/_src/irreps_array_test.py
similarity index 100%
rename from e3nn_jax/_src/irreps_array_test.py
rename to tests/_src/irreps_array_test.py
diff --git a/e3nn_jax/_src/irreps_test.py b/tests/_src/irreps_test.py
similarity index 100%
rename from e3nn_jax/_src/irreps_test.py
rename to tests/_src/irreps_test.py
diff --git a/e3nn_jax/_src/legacy/core_tensor_product_test.py b/tests/_src/legacy/core_tensor_product_test.py
similarity index 100%
rename from e3nn_jax/_src/legacy/core_tensor_product_test.py
rename to tests/_src/legacy/core_tensor_product_test.py
diff --git a/e3nn_jax/_src/linear_flax_test.py b/tests/_src/linear_flax_test.py
similarity index 100%
rename from e3nn_jax/_src/linear_flax_test.py
rename to tests/_src/linear_flax_test.py
diff --git a/e3nn_jax/_src/linear_haiku_test.py b/tests/_src/linear_haiku_test.py
similarity index 100%
rename from e3nn_jax/_src/linear_haiku_test.py
rename to tests/_src/linear_haiku_test.py
diff --git a/e3nn_jax/_src/perm_test.py b/tests/_src/perm_test.py
similarity index 100%
rename from e3nn_jax/_src/perm_test.py
rename to tests/_src/perm_test.py
diff --git a/e3nn_jax/_src/radial_test.py b/tests/_src/radial_test.py
similarity index 100%
rename from e3nn_jax/_src/radial_test.py
rename to tests/_src/radial_test.py
diff --git a/e3nn_jax/_src/radius_graph_test.py b/tests/_src/radius_graph_test.py
similarity index 100%
rename from e3nn_jax/_src/radius_graph_test.py
rename to tests/_src/radius_graph_test.py
diff --git a/e3nn_jax/_src/reduced_tensor_product_test.py b/tests/_src/reduced_tensor_product_test.py
similarity index 100%
rename from e3nn_jax/_src/reduced_tensor_product_test.py
rename to tests/_src/reduced_tensor_product_test.py
diff --git a/e3nn_jax/_src/rotation_test.py b/tests/_src/rotation_test.py
similarity index 100%
rename from e3nn_jax/_src/rotation_test.py
rename to tests/_src/rotation_test.py
diff --git a/e3nn_jax/_src/s2grid_test.py b/tests/_src/s2grid_test.py
similarity index 100%
rename from e3nn_jax/_src/s2grid_test.py
rename to tests/_src/s2grid_test.py
diff --git a/e3nn_jax/_src/scatter_test.py b/tests/_src/scatter_test.py
similarity index 100%
rename from e3nn_jax/_src/scatter_test.py
rename to tests/_src/scatter_test.py
diff --git a/e3nn_jax/_src/so3_test.py b/tests/_src/so3_test.py
similarity index 100%
rename from e3nn_jax/_src/so3_test.py
rename to tests/_src/so3_test.py
diff --git a/e3nn_jax/_src/spherical_harmonics/sh_test.py b/tests/_src/spherical_harmonics/sh_test.py
similarity index 100%
rename from e3nn_jax/_src/spherical_harmonics/sh_test.py
rename to tests/_src/spherical_harmonics/sh_test.py
diff --git a/e3nn_jax/_src/symmetric_tensor_product_haiku_test.py b/tests/_src/symmetric_tensor_product_haiku_test.py
similarity index 100%
rename from e3nn_jax/_src/symmetric_tensor_product_haiku_test.py
rename to tests/_src/symmetric_tensor_product_haiku_test.py
diff --git a/e3nn_jax/_src/tensor_products_test.py b/tests/_src/tensor_products_test.py
similarity index 100%
rename from e3nn_jax/_src/tensor_products_test.py
rename to tests/_src/tensor_products_test.py
diff --git a/e3nn_jax/_src/utils/einsum_test.py b/tests/_src/utils/einsum_test.py
similarity index 100%
rename from e3nn_jax/_src/utils/einsum_test.py
rename to tests/_src/utils/einsum_test.py
diff --git a/e3nn_jax/_src/utils/jit_test.py b/tests/_src/utils/jit_test.py
similarity index 100%
rename from e3nn_jax/_src/utils/jit_test.py
rename to tests/_src/utils/jit_test.py
diff --git a/conftest.py b/tests/conftest.py
similarity index 100%
rename from conftest.py
rename to tests/conftest.py
diff --git a/e3nn_jax/experimental/linear_shtp_test.py b/tests/experimental/linear_shtp_test.py
similarity index 100%
rename from e3nn_jax/experimental/linear_shtp_test.py
rename to tests/experimental/linear_shtp_test.py
diff --git a/e3nn_jax/experimental/point_convolution_test.py b/tests/experimental/point_convolution_test.py
similarity index 100%
rename from e3nn_jax/experimental/point_convolution_test.py
rename to tests/experimental/point_convolution_test.py
diff --git a/e3nn_jax/experimental/transformer_test.py b/tests/experimental/transformer_test.py
similarity index 100%
rename from e3nn_jax/experimental/transformer_test.py
rename to tests/experimental/transformer_test.py
diff --git a/e3nn_jax/experimental/voxel_convolution_test.py b/tests/experimental/voxel_convolution_test.py
similarity index 100%
rename from e3nn_jax/experimental/voxel_convolution_test.py
rename to tests/experimental/voxel_convolution_test.py
diff --git a/e3nn_jax/experimental/voxel_pooling_test.py b/tests/experimental/voxel_pooling_test.py
similarity index 100%
rename from e3nn_jax/experimental/voxel_pooling_test.py
rename to tests/experimental/voxel_pooling_test.py
diff --git a/noxfile.py b/tests/noxfile.py
similarity index 100%
rename from noxfile.py
rename to tests/noxfile.py