diff --git a/.github/workflows/test_python_cplusplus.yml b/.github/workflows/test_python_cplusplus.yml
index 23337986db..6c5e1bd7b7 100644
--- a/.github/workflows/test_python_cplusplus.yml
+++ b/.github/workflows/test_python_cplusplus.yml
@@ -231,11 +231,6 @@ jobs:
     - name: Install python package
       run: scripts/installAmiciSource.sh
 
-    - name: Install notebook dependencies
-      run: |
-        source venv/bin/activate \
-        && pip install jax[cpu]
-
     - name: example notebooks
       run: scripts/runNotebook.sh python/examples/example_*/
 
diff --git a/python/sdist/pyproject.toml b/python/sdist/pyproject.toml
index 22f33eda2c..1d641abf28 100644
--- a/python/sdist/pyproject.toml
+++ b/python/sdist/pyproject.toml
@@ -74,14 +74,21 @@ test = [
     "scipy",
     "pooch"
 ]
-vis =[
+vis = [
     "matplotlib",
     "seaborn",
 ]
-examples =[
+examples = [
     "jupyter",
     "scipy",
 ]
+jax = [
+    "jax>=0.4.34",
+    "jaxlib>=0.4.34",
+    "diffrax>=0.6.0",
+    "equinox>=0.11.8",
+    "optimistix>=0.0.8",
+]
 
 [project.scripts]
 # amici_import_petab.py is kept for backwards compatibility
diff --git a/python/sdist/setup.cfg b/python/sdist/setup.cfg
deleted file mode 100644
index 97e5681c3b..0000000000
--- a/python/sdist/setup.cfg
+++ /dev/null
@@ -1,97 +0,0 @@
-[metadata]
-name = amici
-description = Advanced multi-language Interface to CVODES and IDAS
-version = file: amici/version.txt
-license = BSD 3-Clause License
-url = https://github.com/AMICI-dev/AMICI
-keywords = differential equations, simulation, ode, cvodes, systems biology, sensitivity analysis, sbml, pysb, petab
-author = Fabian Froehlich, Jan Hasenauer, Daniel Weindl and Paul Stapor
-author_email = fabian_froehlich@hms.harvard.edu
-project_urls =
-    Bug Reports = https://github.com/AMICI-dev/AMICI/issues
-    Source = https://github.com/AMICI-dev/AMICI
-    Documentation = https://amici.readthedocs.io/en/latest/
-classifiers =
-    Development Status :: 5 - Production/Stable
-    Intended Audience :: Science/Research
-    License :: OSI Approved :: BSD License
-    Operating System :: POSIX :: Linux
-    Operating System :: MacOS :: MacOS X
-    Programming Language :: Python
-    Programming Language :: C++
-    Topic :: Scientific/Engineering :: Bio-Informatics
-
-[options]
-packages = find_namespace:
-package_dir =
-    amici = amici
-python_requires = >=3.9
-install_requires =
-    cmake-build-extension==0.5.1
-    sympy>=1.9
-    numpy>=1.19.3; python_version=='3.9'
-    numpy>=1.21.4; python_version>='3.10'
-    numpy>=1.23.2; python_version=='3.11'
-    numpy; python_version>='3.12'
-    python-libsbml
-    pandas>=2.0.2
-    pyarrow
-    wurlitzer
-    toposort
-    setuptools>=48
-    mpmath
-include_package_data = True
-zip_safe = False
-
-[options.extras_require]
-# Don't include any URLs here - they are not supported by PyPI:
-#  HTTPError: 400 Bad Request from https://upload.pypi.org/legacy/
-#  Invalid value for requires_dist. Error: Can't have direct dependency: ...
-petab = petab>=0.2.9
-pysb = pysb>=1.13.1
-jax =
-    jax
-    jaxlib
-    diffrax
-    equinox
-    optimistix
-test =
-    benchmark_models_petab @ git+https://github.com/Benchmarking-Initiative/Benchmark-Models-PEtab.git@master#subdirectory=src/python
-    h5py
-    pytest
-    pytest-cov
-    pytest-rerunfailures
-    coverage
-    shyaml
-    antimony>=2.13
-    # see https://github.com/sys-bio/antimony/issues/92
-    #  unsupported x86_64 / x86_64h
-    antimony!=2.14; platform_system=='Darwin' and platform_machine in 'x86_64h'
-    scipy
-    pooch
-vis =
-    matplotlib
-    seaborn
-examples =
-    jupyter
-    scipy
-
-[options.package_data]
-amici =
-    amici/include/amici/*
-    src/*template*
-    swig/*
-    libs/*
-    setup.py.template
-
-[options.exclude_package_data]
-* =
-    README.txt
-
-
-[options.entry_points]
-
-; amici_import_petab.py is kept for backwards compatibility
-console_scripts =
-    amici_import_petab = amici.petab.cli.import_petab:_main
-    amici_import_petab.py = amici.petab.cli.import_petab:_main