Fix pfft gradients (#34)

* update jaxdecomp version and test gradients

* Prepare for DTO tests

* format
This commit is contained in:
Wassim KABALAN 2024-12-22 18:47:42 +01:00 committed by GitHub
parent d81a2529e7
commit cb2a7ab17f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 88 additions and 1 deletions

View file

@ -11,7 +11,7 @@ readme = "README.md"
requires-python = ">=3.9"
license = { file = "LICENSE" }
urls = { "Homepage" = "https://github.com/DifferentiableUniverseInitiative/JaxPM" }
dependencies = ["jax_cosmo", "jax>=0.4.30", "jaxdecomp>=0.2.2"]
dependencies = ["jax_cosmo", "jax>=0.4.35", "jaxdecomp>=0.2.3"]
[tool.setuptools]
packages = ["jaxpm"]