diff --git a/.gitignore b/.gitignore
index f8cc725..f53c4ff 100644
--- a/.gitignore
+++ b/.gitignore
@@ -29,6 +29,7 @@ share/python-wheels/
 .installed.cfg
 *.egg
 MANIFEST
+jax_scalify/version.py
 
 # PyInstaller
 #  Usually these files are written by a python script from a template
diff --git a/jax_scalify/__init__.py b/jax_scalify/__init__.py
index 32988d7..7b54bf3 100644
--- a/jax_scalify/__init__.py
+++ b/jax_scalify/__init__.py
@@ -1,6 +1,5 @@
 # Copyright (c) 2023 Graphcore Ltd. All rights reserved.
 from . import core, lax, ops, tree
-from ._version import __version__
 from .core import (  # noqa: F401
     Pow2RoundMode,
     ScaledArray,
@@ -11,3 +10,4 @@
     scaled_array,
     scalify,
 )
+from .version import __version__
diff --git a/jax_scalify/_version.py b/jax_scalify/_version.py
deleted file mode 100644
index 64b95fd..0000000
--- a/jax_scalify/_version.py
+++ /dev/null
@@ -1,2 +0,0 @@
-# Copyright (c) 2023 Graphcore Ltd. All rights reserved.
-__version__ = "0.1"
diff --git a/pyproject.toml b/pyproject.toml
index d9130e6..e9f9cd6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,7 +7,6 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jax_scalify"
-version = "0.1"
 description="JAX Scalify: end-to-end scaled arithmetic."
 readme = "README.md"
 authors = [
@@ -30,6 +29,7 @@ dependencies = [
   "ml_dtypes",
   "numpy >= 1.22.4"
 ]
+dynamic = ["version"]
 
 [project.urls]
 "Homepage" = "https://github.com/graphcore-research/jax-scalify/#readme"
@@ -43,10 +43,16 @@ test = ["pytest"]
 # Relying on the default setuptools.
 # In case of an issue, can use the following options
 # [tool.setuptools]
-# packages = ["jax_scalify", "jax_scalify.core", "jax_scalify.lax", "jax_scalify.ops"]
+# packages = ["jax_scalify", "jax_scalify.core", "jax_scalify.lax", "jax_scalify.ops", "jax_scalify.tree"]
 # [tool.setuptools.packages]
 # find = {namespaces = false}
 
+[tool.setuptools.dynamic]
+version = {attr = "jax_scalify.version.__version__"}
+
+[tool.setuptools_scm]
+version_file = "jax_scalify/version.py"
+
 [tool.pytest.ini_options]
 minversion = "6.0"
 addopts = ["-ra", "--showlocals", "--strict-config", "-p no:hypothesispytest"]