mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-04-12 06:00:54 +00:00
26 lines
775 B
TOML
26 lines
775 B
TOML
[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"
|
|
]
|