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

17 commits

Author SHA1 Message Date
Wassim KABALAN
2f509932f5 update demo 2024-10-22 12:16:28 -04:00
Wassim KABALAN
4d944f01f2 update demo script 2024-10-22 12:10:18 -04:00
Wassim KABALAN
a160a3faa9 update to leapfrog 2024-10-22 09:03:41 -04:00
Wassim KABALAN
375f2048e4 make distributed pm work in single controller 2024-10-21 14:03:48 -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
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
1f2035176f
Update scripts/distributed_pm.py 
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
 2024-07-19 19:25:26 +02: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
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