[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"
]