guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-29 16:41:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
149 commits 6 branches 6 tags 175 MiB
Commit graph

149 commits

This branch
This branch
All branches
Author SHA1 Message Date
Francois Lanusse
1fcfc2aef2 Merge branch 'main' into 41-spherical-lensing 2025-06-28 23:12:39 +02:00
Wassim KABALAN
6693e5c725
Fix sharding error (#37)
Some checks failed
Code Formatting / formatting (push) Failing after 4m30s
Details
Tests / run_tests (3.10) (push) Failing after 1m41s
Details
Tests / run_tests (3.11) (push) Failing after 1m42s
Details
Tests / run_tests (3.12) (push) Failing after 1m15s
Details 
* Use cosmo as arg for the ODE function

* Update examples

* format

* notebook update

* fix tests

* add correct annotations for weights in painting and warning for cic_paint in distributed pm

* update test_against_fpm

* update distributed tests and add jacfwd jacrev and vmap tests

* format

* add Caveats to notebook readme

* final touches

* update Growth.py to allow using FastPM solver

* fix 2D painting when input is (X , Y , 2) shape

* update cic read halo size and notebooks examples

* Allow env variable control of caching in growth

* Format

* update test jax version

* update notebooks/03-MultiGPU_PM_Halo.ipynb

* update numpy install in wf

* update tolerance :)

* reorganize install in test workflow

* update tests

* add mpi4py

* update tests.yml

* update tests

* update wf

* format

* make normal_field signature consistent with jax.random.normal

* update by default normal_field dtype to match JAX

* format

* debug test workflow

* format

* debug test workflow

* updating tests

* fix accuracy

* fixed tolerance

* adding caching

* Update conftest.py

* Update tolerance and precision settings in distributed PM tests

* revererting back changes to growth.py

---------

Co-authored-by: Francois Lanusse <fr.eiffel@gmail.com>
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
 2025-06-28 23:07:31 +02:00
Wassim Kabalan
f6d547e31f Add Spherical lensing example 2025-06-28 19:25:14 +02:00
Wassim Kabalan
2d21985279 Update tolerance and precision settings in distributed PM tests 2025-06-28 12:08:16 +02:00
Francois Lanusse
e666aada42
Update conftest.py 2025-06-28 02:57:20 +02:00
Francois Lanusse
844cace3d7 adding caching 2025-06-28 02:41:16 +02:00
Francois Lanusse
37976c70d8 fixed tolerance 2025-06-28 02:36:37 +02:00
Francois Lanusse
0c75158043 fix accuracy 2025-06-28 02:21:19 +02:00
Francois Lanusse
627be7a764 updating tests 2025-06-28 01:36:17 +02:00
Wassim Kabalan
7623e60581 debug test workflow 2025-06-12 15:28:03 +02:00
Wassim Kabalan
d874790543 format 2025-06-12 15:20:20 +02:00
Wassim Kabalan
187cf5c4ba debug test workflow 2025-06-12 15:19:36 +02:00
Wassim Kabalan
67a80e1041 format 2025-06-12 14:51:24 +02:00
Wassim Kabalan
e1a8134b8e update by default normal_field dtype to match JAX 2025-06-12 14:46:27 +02:00
Wassim Kabalan
f391a79211 make normal_field signature consistent with jax.random.normal 2025-06-09 19:35:40 +02:00
Wassim Kabalan
5807e1d3f4 format 2025-06-08 15:37:06 +02:00
Wassim Kabalan
a7fcba0e1f update wf 2025-06-08 15:35:57 +02:00
Wassim Kabalan
c6a7dd4e4e update tests 2025-06-08 15:27:32 +02:00
Wassim Kabalan
e0ba85fb58 update tests.yml 2025-06-08 12:03:44 +02:00
Wassim Kabalan
12eddc4e6a add mpi4py 2025-06-08 11:48:11 +02:00
Wassim Kabalan
0eb4c371e3 update tests 2025-06-08 11:43:15 +02:00
Wassim Kabalan
3be619a2db reorganize install in test workflow 2025-06-08 11:36:06 +02:00
Wassim Kabalan
f0b849cf5f update tolerance :) 2025-06-08 11:35:57 +02:00
Wassim Kabalan
995cc4c78c update numpy install in wf 2025-06-08 11:26:32 +02:00
Wassim Kabalan
7b7205e3b3 update notebooks/03-MultiGPU_PM_Halo.ipynb 2025-06-08 11:25:51 +02:00
Wassim Kabalan
6aacd81bd6 update test jax version 2025-06-08 10:56:34 +02:00
Wassim Kabalan
49c93aacf6 Format 2025-06-08 10:45:20 +02:00
Wassim Kabalan
41ae41ace3 Allow env variable control of caching in growth 2025-06-08 10:45:04 +02:00
Wassim Kabalan
e7112e0c25 update cic read halo size and notebooks examples 2025-06-07 19:26:37 +02:00
Wassim Kabalan
d4049e5db4 fix 2D painting when input is (X , Y , 2) shape 2025-05-09 22:07:53 +02:00
Wassim Kabalan
20ace41d32 update Growth.py to allow using FastPM solver 2025-05-08 15:08:28 +02:00
Wassim Kabalan
e1daa8cba4 final touches 2025-02-28 14:03:33 +01:00
Wassim Kabalan
4e4d3745f0 add Caveats to notebook readme 2025-02-28 13:48:08 +01:00
Wassim Kabalan
9f494da317 format 2025-02-28 13:47:43 +01:00
Wassim Kabalan
580387ce1c update distributed tests and add jacfwd jacrev and vmap tests 2025-02-28 13:47:22 +01:00
Wassim Kabalan
eb0778aa24 update test_against_fpm 2025-02-28 13:47:00 +01:00
Wassim Kabalan
91d3292923 add correct annotations for weights in painting and warning for cic_paint in distributed pm 2025-02-28 13:46:41 +01:00
Wassim Kabalan
f8325b1c67 fix tests 2025-02-28 10:06:09 +01:00
Wassim Kabalan
f3b8f4160e notebook update 2025-02-28 09:59:04 +01:00
Wassim Kabalan
b43cb373a0 format 2025-02-28 09:56:00 +01:00
Wassim Kabalan
51ee4dd937 update examples 2025-02-28 09:54:53 +01:00
Wassim Kabalan
8e0f300572 Update examples 2025-02-26 14:32:47 +01:00
Wassim Kabalan
0b08c6f59a Use cosmo as arg for the ODE function 2025-02-26 14:05:25 +01:00
Wassim KABALAN
cb2a7ab17f
Fix pfft gradients (#34) 
* update jaxdecomp version and test gradients

* Prepare for DTO tests

* format
 2024-12-22 12:47:42 -05:00
Francois Lanusse
d81a2529e7 minor typo fix 2024-12-21 15:28:20 -05:00
Francois Lanusse
15f2fb1ee6 adding notice 2024-12-21 15:26:53 -05:00
Francois Lanusse
ae0f439ae4 fixing formatting of notebook 2024-12-21 13:14:42 -05:00
Francois Lanusse
ea9fbf6aa8
Update README.md 2024-12-21 13:13:37 -05:00
Francois Lanusse
ad16a0659a Created using Colab 2024-12-21 13:10:15 -05:00
Francois Lanusse
f245a1f685
Pypi upload compatible version (#33) 
* moving test dependencies separately

* adding manifest to remove unecessary files

* updating name of project

* Fixing formatting

* Adding badge for pypi version

* Adding very simple install instructions
 2024-12-21 11:47:13 -05:00