JaxPM

guilhem_lavaux/JaxPM

Fork 1

mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-07-19 18:03:06 +00:00

Commit graph

bfb77810fa

Merge bfc9c775f8 into 7d76573701 Wassim KABALAN 2025-07-08 08:55:19 +00:00
bfc9c775f8 update 41-spherical-lensing Wassim Kabalan 2025-07-08 10:55:07 +02:00
3522edb6b1 update Wassim Kabalan 2025-07-07 12:09:51 +02:00
b308f9a80f update Wassim Kabalan 2025-07-07 12:09:32 +02:00
b0367e2ee2 update Wassim Kabalan 2025-07-05 23:34:59 +02:00
4df132ed29 update Wassim Kabalan 2025-07-05 22:25:30 +02:00
045579dede stash Wassim Kabalan 2025-07-05 12:01:51 +02:00
2da4962732 update Wassim Kabalan 2025-07-04 22:32:22 +02:00
7962440619

Merge 60f16a9180 into 7d76573701 Wassim KABALAN 2025-07-04 19:32:53 +02:00
60f16a9180 format 45-differentiable-painting-strategy-for-density-shells Wassim Kabalan 2025-07-04 19:32:42 +02:00
c485a3aed0 added a prototype of RBF kernel Wassim Kabalan 2025-07-04 19:31:23 +02:00
b2075bba98 add healpix spherical CIC kernel Wassim Kabalan 2025-07-04 15:03:11 +02:00
c0e93d9e1a implementing distributed spherical lensing Wassim Kabalan 2025-07-04 00:55:52 +02:00
8d2560e30c update Wassim Kabalan 2025-07-04 00:34:11 +02:00
ea7de7c9d0 update notebooks Wassim Kabalan 2025-07-04 00:22:34 +02:00
87fb557e10 update flat Sky notebook Wassim Kabalan 2025-07-03 23:56:49 +02:00
73ec466ab3 format Wassim Kabalan 2025-07-03 22:56:59 +02:00
b941194ce8 Update requirements-test.txt to include lenstools Wassim Kabalan 2025-07-03 22:54:58 +02:00
928d715c11 add tests for flat_sky_cl_square Wassim Kabalan 2025-07-03 22:53:58 +02:00
7e7ecc35a2 Add flat_sky_cl function to utils.py Wassim Kabalan 2025-07-03 22:51:42 +02:00
ad16e04a2c update Wassim Kabalan 2025-07-03 22:37:44 +02:00
c71c44fc4e update Wassim Kabalan 2025-07-03 21:53:11 +02:00
a18ce8027d add spherical RayTracing Wassim Kabalan 2025-07-03 21:51:22 +02:00
35173820b7 updating spherical ray tracing and power spectrum analysis scripts Wassim Kabalan 2025-07-01 11:32:09 +02:00
dc5371e56e update ray tracing Wassim Kabalan 2025-06-30 13:02:23 +02:00
09d82283f5 Merge remote-tracking branch 'origin/main' into 41-spherical-lensing Wassim Kabalan 2025-06-29 11:32:30 +02:00
7d76573701

Replace np.power 0.5 by np.sqrt (#43) main v0.1.6 Wassim KABALAN 2025-06-29 10:51:45 +02:00
9d0b047f06 fixing formatting Francois Lanusse 2025-06-28 23:21:31 +02:00
1fcfc2aef2 Merge branch 'main' into 41-spherical-lensing Francois Lanusse 2025-06-28 23:12:39 +02:00
475072ce64 Merge branch 'main' into 38-replace-nppower-by-npsqrt Francois Lanusse 2025-06-28 23:09:09 +02:00
6693e5c725

Fix sharding error (#37) Wassim KABALAN 2025-06-28 23:07:31 +02:00
77ea494e4f revererting back changes to growth.py Francois Lanusse 2025-06-28 22:02:14 +02:00
6de6ded383 format Wassim Kabalan 2025-06-28 19:37:15 +02:00
14a3899d67 Switch **2 by np.sqrt Wassim Kabalan 2025-06-28 19:35:10 +02:00
f6d547e31f Add Spherical lensing example Wassim Kabalan 2025-06-28 19:25:14 +02:00
2d21985279 Update tolerance and precision settings in distributed PM tests Wassim Kabalan 2025-06-28 12:08:16 +02:00
e666aada42

Update conftest.py Francois Lanusse 2025-06-28 02:57:20 +02:00
844cace3d7 adding caching Francois Lanusse 2025-06-28 02:41:16 +02:00
37976c70d8 fixed tolerance Francois Lanusse 2025-06-28 02:36:37 +02:00
0c75158043 fix accuracy Francois Lanusse 2025-06-28 02:21:19 +02:00
627be7a764 updating tests Francois Lanusse 2025-06-28 01:36:17 +02:00
7623e60581 debug test workflow Wassim Kabalan 2025-06-12 15:28:03 +02:00
d874790543 format Wassim Kabalan 2025-06-12 15:20:20 +02:00
187cf5c4ba debug test workflow Wassim Kabalan 2025-06-12 15:19:36 +02:00
67a80e1041 format Wassim Kabalan 2025-06-12 14:51:24 +02:00
e1a8134b8e update by default normal_field dtype to match JAX Wassim Kabalan 2025-06-12 14:46:27 +02:00
cf979f3250 ? chenze@igpu 2025-06-10 22:46:26 +09:00
f391a79211 make normal_field signature consistent with jax.random.normal Wassim Kabalan 2025-06-09 19:35:40 +02:00
5807e1d3f4 format Wassim Kabalan 2025-06-08 15:37:06 +02:00
a7fcba0e1f update wf Wassim Kabalan 2025-06-08 15:35:57 +02:00
c6a7dd4e4e update tests Wassim Kabalan 2025-06-08 15:27:32 +02:00
e0ba85fb58 update tests.yml Wassim Kabalan 2025-06-08 12:03:44 +02:00
12eddc4e6a add mpi4py Wassim Kabalan 2025-06-08 11:48:11 +02:00
0eb4c371e3 update tests Wassim Kabalan 2025-06-08 11:43:15 +02:00
3be619a2db reorganize install in test workflow Wassim Kabalan 2025-06-08 11:36:06 +02:00
f0b849cf5f update tolerance :) Wassim Kabalan 2025-06-08 11:35:57 +02:00
995cc4c78c update numpy install in wf Wassim Kabalan 2025-06-08 11:26:32 +02:00
7b7205e3b3 update notebooks/03-MultiGPU_PM_Halo.ipynb Wassim Kabalan 2025-06-08 11:25:51 +02:00
6aacd81bd6 update test jax version Wassim Kabalan 2025-06-08 10:56:34 +02:00
49c93aacf6 Format Wassim Kabalan 2025-06-08 10:45:20 +02:00
41ae41ace3 Allow env variable control of caching in growth Wassim Kabalan 2025-06-08 10:45:04 +02:00
e7112e0c25 update cic read halo size and notebooks examples Wassim Kabalan 2025-06-07 19:26:37 +02:00
d4049e5db4 fix 2D painting when input is (X , Y , 2) shape Wassim Kabalan 2025-05-09 22:07:53 +02:00
20ace41d32 update Growth.py to allow using FastPM solver Wassim Kabalan 2025-05-08 15:08:28 +02:00
e1daa8cba4 final touches Wassim Kabalan 2025-02-28 14:03:33 +01:00
4e4d3745f0 add Caveats to notebook readme Wassim Kabalan 2025-02-28 13:48:08 +01:00
9f494da317 format Wassim Kabalan 2025-02-28 13:47:43 +01:00
580387ce1c update distributed tests and add jacfwd jacrev and vmap tests Wassim Kabalan 2025-02-28 13:47:22 +01:00
eb0778aa24 update test_against_fpm Wassim Kabalan 2025-02-28 13:47:00 +01:00
91d3292923 add correct annotations for weights in painting and warning for cic_paint in distributed pm Wassim Kabalan 2025-02-28 13:46:41 +01:00
f8325b1c67 fix tests Wassim Kabalan 2025-02-28 10:06:09 +01:00
f3b8f4160e notebook update Wassim Kabalan 2025-02-28 09:59:04 +01:00
b43cb373a0 format Wassim Kabalan 2025-02-28 09:56:00 +01:00
51ee4dd937 update examples Wassim Kabalan 2025-02-28 09:54:53 +01:00
8e0f300572 Update examples Wassim Kabalan 2025-02-26 14:32:47 +01:00
0b08c6f59a Use cosmo as arg for the ODE function Wassim Kabalan 2025-02-26 14:05:25 +01:00
9b8e42aa00 updates for FullField Wassim Kabalan 2025-02-05 14:26:28 +01:00
34bf577b1b small fix Wassim Kabalan 2025-01-21 12:08:47 +01:00
d4f72c8244 format Wassim Kabalan 2025-01-21 11:21:35 +01:00
6aa2ccf7b5 don't send static cosmo into ODETerm Wassim Kabalan 2025-01-21 11:18:00 +01:00
4ac82abe39 remove extra text Wassim Kabalan 2025-01-21 10:15:11 +01:00
6ab26ea1ec remove extra text Wassim Kabalan 2025-01-21 10:15:11 +01:00
0b723c21ae bump minimum jd version Wassim Kabalan 2025-01-20 23:08:12 +01:00
03610602a8 bump minimum jd version Wassim Kabalan 2025-01-20 23:08:12 +01:00
a8f11a75ea format Wassim Kabalan 2025-01-20 22:47:07 +01:00
bf12da3bb7 Merge remote-tracking branch 'origin/main' into fix-pfft-gradients Wassim Kabalan 2025-01-20 22:45:32 +01:00
1f5c619531 format Wassim Kabalan 2025-01-20 22:41:19 +01:00
20fe25c324 update tests and add test for FWD REV gradient Wassim Kabalan 2025-01-20 22:40:28 +01:00
151fa09247 power spec should accept pytrees Wassim Kabalan 2025-01-20 22:40:01 +01:00
38f6599974 painting now accepts pytrees Wassim Kabalan 2025-01-20 22:39:40 +01:00
9e203b5680 pm now accept pytrees Wassim Kabalan 2025-01-18 01:13:51 +01:00
204a9526ec kernels now accept pytrees Wassim Kabalan 2025-01-18 01:13:39 +01:00
f5755b4b5d update painting functions to accept pytrees Wassim Kabalan 2025-01-18 01:13:24 +01:00
7c3577ea71 update distributed ops for new jd Wassim Kabalan 2025-01-18 01:12:53 +01:00
cb2a7ab17f

Fix pfft gradients (#34) Wassim KABALAN 2024-12-22 18:47:42 +01:00
bbacd45dcf format Wassim Kabalan 2024-12-21 23:27:05 +01:00
a924458f0d Prepare for DTO tests Wassim Kabalan 2024-12-21 23:24:19 +01:00
b132a0e2aa update jaxdecomp version and test gradients Wassim Kabalan 2024-12-21 23:14:45 +01:00
45aa70899a adding more information in contributing additional_testing Francois Lanusse 2024-12-21 16:29:51 -05:00
e316d8a0b2 adding caching of dependencies Francois Lanusse 2024-12-21 15:57:40 -05:00