diff --git a/README.md b/README.md
index 6463b79..5249548 100644
--- a/README.md
+++ b/README.md
@@ -183,6 +183,10 @@ source $PATH_TO_LAGRANGEBENCH_VENV/bin/activate
 # install pre-commit hooks defined in .pre-commit-config.yaml
 # ruff is configured in pyproject.toml
 pre-commit install
+
+# if you want to bump the version in both pyproject.toml and __init__.py, do
+poetry self add poetry-bumpversion
+poetry version patch  # or minor/major
 ```
 
 After you have run `git add <FILE>` and try to `git commit`, the pre-commit hook will
@@ -201,6 +205,7 @@ LagrangeBench can be installed by cloning the repository or as a standalone libr
 - Configs in `configs/` and `lagrangebench/defaults.py`
 - Zenodo URLs in `download_data.sh` and `lagrangebench/data/data.py`
 - Dependencies in `pyproject.toml`, `requirements_cuda.txt`, and `docs/requirements.txt`
+- Library version in `pyproject.toml` and `lagrangebench/__init__.py`
 
 
 ## Citation
diff --git a/poetry.lock b/poetry.lock
index a0666e6..10bd272 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -3164,17 +3164,6 @@ files = [
 ml-dtypes = ">=0.3.1"
 numpy = ">=1.16.0"
 
-[[package]]
-name = "toml"
-version = "0.10.2"
-description = "Python Library for Tom's Obvious, Minimal Language"
-optional = false
-python-versions = ">=2.6, !=3.0.*, !=3.1.*, !=3.2.*"
-files = [
-    {file = "toml-0.10.2-py2.py3-none-any.whl", hash = "sha256:806143ae5bfb6a3c6e736a764057db0e6a0e05e338b5630894a5f779cabb4f9b"},
-    {file = "toml-0.10.2.tar.gz", hash = "sha256:b3bda1d108d5dd99f4a20d24d9c348e91c4db7ab1b749200bded2f839ccbe68f"},
-]
-
 [[package]]
 name = "tomli"
 version = "2.0.1"
@@ -3449,4 +3438,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<=3.11"
-content-hash = "a248a6e4f676b7a384560f80f41db5b185520ca822ab8b849586bc51c2ace946"
+content-hash = "897da048fa53886069654ac3de3780965b06195533e9e92f0332748fddcb28e7"
diff --git a/pyproject.toml b/pyproject.toml
index 00d6f4d..1129ee1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -55,7 +55,6 @@ matscipy = "^0.8.0"
 torch = {version = "2.1.0+cpu", source = "torchcpu"}
 wget = "^3.2"
 omegaconf = "^2.3.0"
-toml = "^0.10.2"
 
 [tool.poetry.group.dev.dependencies]
 # mypy = ">=1.8.0" - consider in the future
@@ -106,6 +105,7 @@ filterwarnings = [
     "ignore::DeprecationWarning:^(?!.*lagrangebench).*"
 ]
 
+[tool.poetry_bumpversion.file."lagrangebench/__init__.py"]
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
diff --git a/requirements_cuda.txt b/requirements_cuda.txt
index abb46bb..535f6eb 100644
--- a/requirements_cuda.txt
+++ b/requirements_cuda.txt
@@ -16,7 +16,6 @@ optax>=0.1.7
 ott-jax>=0.4.2
 pyvista
 PyYAML
-toml>=0.10.2
 torch==2.1.0+cpu
 wandb
 wget