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
109 commits 7 branches 5 tags 164 MiB
Commit graph

10 commits

Author SHA1 Message Date
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