guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-04-07 12:20:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
182 commits 7 branches 5 tags 164 MiB
Commit graph

39 commits

Author SHA1 Message Date
Wassim Kabalan
af29c4005d Format 2024-12-08 22:45:09 +01:00
Wassim Kabalan
21373b89ee update code 2024-12-06 18:56:24 +01:00
Wassim Kabalan
e0c118a540 By default use absoulute painting with 2024-12-05 18:23:18 +01:00
Wassim KABALAN
19011d0712 apply formating 2024-10-27 03:50:34 +01:00
Wassim KABALAN
d62c38f457 fix code in LPT2 2024-10-27 03:48:38 +01:00
Francois Lanusse
0f833f0cb4 remove deprecated stuff 2024-10-26 22:48:38 +02:00
Wassim KABALAN
5d4f438e92 update PGDCorrection and neural ode to use new fft3d 2024-10-25 10:14:29 +02:00
Wassim KABALAN
31ca41b0a7 remove duplicate get_ode_fn 2024-10-22 13:38:16 -04:00
Wassim KABALAN
d28982eec7 Small fix 2024-10-22 13:00:25 -04:00
Wassim KABALAN
85cca44fb0 Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto 2024-10-22 12:59:50 -04:00
Wassim KABALAN
82b8f563a0 format 2024-10-22 12:57:34 -04:00
Wassim KABALAN
86233081e2 merge hugos LPT2 code 2024-10-22 12:57:05 -04:00
Wassim KABALAN
80c56dced5 pm.py no longer uses global mesh 2024-10-22 12:09:50 -04:00
Wassim KABALAN
01b952701e update for latest jaxDecomp 2024-10-21 13:55:48 -04:00
Wassim KABALAN
8da3149581 fix formatting from main 2024-08-03 00:45:35 +02:00
Wassim KABALAN
783a97423e Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto 2024-08-03 00:23:54 +02:00
Wassim KABALAN
ece8c93540 format 2024-08-03 00:23:40 +02:00
Wassim KABALAN
ccbfee3615 add assert to make pyright happy 2024-08-02 21:21:59 +02:00
Hugo Simonfroy
d85867e88d 2lpt, get_ode, invlaplace, docstrings 2024-07-31 00:46:53 +02:00
Francois Lanusse
9cbdf18932
Update jaxpm/pm.py 2024-07-19 10:49:51 -04:00
Francois Lanusse
945dae4f80
Merge branch 'main' into neural_ode 2024-07-19 10:48:09 -04:00
Wassim KABALAN
ed8cf8e532 add lpt2 2024-07-18 17:04:05 +02:00
Wassim KABALAN
4f508b7cb6 format 2024-07-18 13:16:14 +02:00
Wassim KABALAN
7f48cfa8af Fix seed for distributed normal 2024-07-18 12:43:24 +02:00
Wassim KABALAN
7501b5bc6d Use new cic_paint with halo 2024-07-18 12:42:07 +02:00
EiffL
319942a6bc update formatting 2024-07-09 18:02:57 -04:00
EiffL
6408aff1de put back old functgion 2024-07-09 18:00:35 -04:00
EiffL
a742065ffd adding example of distributed solution 2024-07-09 17:45:28 -04:00
EiffL
a2811c0606 Applying formatting 2024-07-09 14:54:34 -04:00
denise lanzieri
b26cfcd01a creoss correlation function 2022-06-18 18:23:46 +02:00
denise lanzieri
f185b8b913 few adjustments to PGD correction 2022-06-13 17:17:19 +02:00
denise lanzieri
5b72a73448 neural ode added 2022-06-11 14:28:30 +02:00
Denise Lanzieri
5394abbed1
Update jaxpm/pm.py 
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
 2022-05-18 09:59:59 +02:00
Denise Lanzieri
782c9dbc6c
Update jaxpm/pm.py 
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
 2022-05-18 09:58:46 +02:00
Denise Lanzieri
8266f11369
Update jaxpm/pm.py 
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
 2022-05-18 09:58:34 +02:00
Denise Lanzieri
b737631346
Update jaxpm/pm.py 
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
 2022-05-18 09:55:31 +02:00
denise lanzieri
d3026f7849 PGD 2022-05-17 15:28:30 +02:00
EiffL
5549568295 fix normalization of init cond 2022-03-25 22:34:13 +01:00
EiffL
25bde5581e Adds demo and notebooks 2022-02-14 01:59:12 +01:00