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

20 commits

Author SHA1 Message Date
Wassim KABALAN
4342279817 apply formatting 2024-10-27 00:52:14 +02:00
Wassim KABALAN
8e8e8964be jaxDecomp pfft3d promotes to complex automatically 2024-10-25 10:15:43 +02:00
Wassim KABALAN
45b2c7f0d2 By default check_rep is false for shard_map 2024-10-22 17:30:56 -04:00
Wassim KABALAN
0433c615f3 quick fix in distributed 2024-10-22 12:15:37 -04:00
Wassim KABALAN
56ffd263f6 quick fix in distributed 2024-10-22 12:07:06 -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
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
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
ece8c93540 format 2024-08-03 00:23:40 +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
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
Wassim KABALAN
ac86468c7c add halo exchange and slice pad 2024-07-10 00:25:52 +02:00
EiffL
319942a6bc update formatting 2024-07-09 18:02:57 -04:00
EiffL
a742065ffd adding example of distributed solution 2024-07-09 17:45:28 -04:00