Upgrade setup.py to pyproject

pyproject.toml

@@ -0,0 +1,26 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "JaxPM"
+version = "0.0.1"
+description = "A dead simple FastPM implementation in JAX"
+authors = [{ name = "JaxPM developers" }]
+readme = "README.md"
+requires-python = ">=3.9"
+license = { file = "LICENSE" }
+homepage = "https://github.com/DifferentiableUniverseInitiative/JaxPM"
+dependencies = ["jax_cosmo", "jax>=0.4.30", "jaxdecomp>=0.2.0"]
+
+[project.optional-dependencies]
+test = [
+    "jax_cosmo",
+    "jaxdecomp>=0.2.0",
+    "pytest>=8.0.0",
+    "jax[cpu]>=0.4.30",
+    "pfft-python @ git+https://github.com/MP-Gadget/pfft-python",
+    "pmesh @ git+https://github.com/ASKabalan/pmesh",
+    "fastpm @ git+https://github.com/ASKabalan/fastpm-python",
+    "diffrax"
+]

setup.py

@@ -1,11 +0,0 @@
-from setuptools import find_packages, setup
-
-setup(
-    name='JaxPM',
-    version='0.0.1',
-    url='https://github.com/DifferentiableUniverseInitiative/JaxPM',
-    author='JaxPM developers',
-    description='A dead simple FastPM implementation in JAX',
-    packages=find_packages(),
-    install_requires=['jax', 'jax_cosmo', 'jaxdecomp'],
-)