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

128 commits

This branch
This branch
All branches
Author SHA1 Message Date
Wassim KABALAN
86233081e2 merge hugos LPT2 code 2024-10-22 12:57:05 -04:00
Wassim KABALAN
2f509932f5 update demo 2024-10-22 12:16:28 -04:00
Wassim KABALAN
0433c615f3 quick fix in distributed 2024-10-22 12:15:37 -04:00
Wassim KABALAN
a8b194f326 quick fix in kernels 2024-10-22 12:15:17 -04:00
Wassim KABALAN
4d944f01f2 update demo script 2024-10-22 12:10:18 -04:00
Wassim KABALAN
105568e8db painting.py no longer uses global mesh 2024-10-22 12:10:01 -04:00
Wassim KABALAN
80c56dced5 pm.py no longer uses global mesh 2024-10-22 12:09:50 -04:00
Wassim KABALAN
56ffd263f6 quick fix in distributed 2024-10-22 12:07:06 -04:00
Wassim KABALAN
a5b267bd63 kernels.py no longer uses global mesh 2024-10-22 11:05:21 -04:00
Wassim KABALAN
591ee32c55 global mesh no longer needed 2024-10-22 11:04:29 -04:00
Wassim KABALAN
38714cf65d add a strict dependency on jaxdecomp 2024-10-22 11:02:55 -04:00
Wassim KABALAN
a160a3faa9 update to leapfrog 2024-10-22 09:03:41 -04:00
Wassim KABALAN
5a587fd402 merge bench_pm 2024-10-21 14:04:09 -04:00
Wassim KABALAN
375f2048e4 make distributed pm work in single controller 2024-10-21 14:03:48 -04:00
Wassim KABALAN
9c94f994ff format 2024-10-21 14:00:14 -04:00
Wassim KABALAN
0ce7219ad8 make normal_field work with single controller 2024-10-21 13:59:55 -04:00
Wassim KABALAN
ff1c5e8362 remove fourrier_space in autoshmap 2024-10-21 13:57:26 -04:00
Wassim KABALAN
01b952701e update for latest jaxDecomp 2024-10-21 13:55:48 -04:00
Wassim KABALAN
afecb13cde revert single halo extent change 2024-10-20 09:58:24 -04:00
Wassim KABALAN
ab86699c88 merge with JZ 2024-10-18 14:59:40 -04:00
Wassim KABALAN
2ea05a1cd6 Update for jaxDecomp pure JAX 2024-10-18 20:45:08 +02:00
Wassim KABALAN
8da3149581 fix formatting from main 2024-08-03 00:45:35 +02:00
Wassim KABALAN
02754cf452 added github workflow 2024-08-03 00:44:05 +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
831291c1f9 bench 2024-08-02 23:39:09 +02:00
Wassim KABALAN
9af4659c81 add PMWD test 2024-08-02 21:22:56 +02:00
Wassim KABALAN
aebc3e72c0 adjust test for hpc-plotter 2024-08-02 21:22:45 +02:00
Wassim KABALAN
ccbfee3615 add assert to make pyright happy 2024-08-02 21:21:59 +02:00
Wassim KABALAN
75604d2396 Shared operation in fourrier space now take inverted sharding axis for 
slabs
 2024-08-02 21:21:06 +02:00
Wassim KABALAN
8c5bd76c33 fix painting issue with slabs 2024-08-02 21:20:15 +02:00
Wassim KABALAN
f25eb7d465 Adjust pencil type for frequencies 2024-07-28 13:44:03 +02:00
Wassim KABALAN
1f2035176f
Update scripts/distributed_pm.py 
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
 2024-07-19 19:25:26 +02:00
Francois Lanusse
2bb0f9e9a9
Merge pull request #22 from DifferentiableUniverseInitiative/all-contributors/add-dlanzieri 
docs: add dlanzieri as a contributor for code
 2024-07-19 10:52:37 -04:00
allcontributors[bot]
b4adc119b6
docs: update .all-contributorsrc [skip ci] 2024-07-19 14:51:26 +00:00
allcontributors[bot]
38270f3358
docs: update README.md [skip ci] 2024-07-19 14:51:25 +00:00
Francois Lanusse
d2ab0fe6e9
Merge pull request #14 from DifferentiableUniverseInitiative/neural_ode 
Fourier-space Neural Network  scheme
 2024-07-19 10:50:36 -04: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
5b7f595c32 Visualize only final field 2024-07-18 17:05:22 +02:00
Wassim KABALAN
0216837033 update benchmark and add slurm 2024-07-18 17:04:52 +02:00
Wassim KABALAN
ed8cf8e532 add lpt2 2024-07-18 17:04:05 +02:00
Wassim KABALAN
1f6b9c3217 times in ms 2024-07-18 13:23:53 +02:00
Wassim KABALAN
5f6d42eaeb add optimised bench script 2024-07-18 13:16:37 +02:00
Wassim KABALAN
4f508b7cb6 format 2024-07-18 13:16:14 +02:00
Wassim KABALAN
82be56836a add example 2024-07-18 12:45:56 +02:00
Wassim KABALAN
abde5439f6 Return normal order frequencies for single GPU 2024-07-18 12:44:18 +02:00
Wassim KABALAN
c81d4d2336 Wrap interpolation function to avoid all gather 2024-07-18 12:43:52 +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