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

25 commits

Author SHA1 Message Date
Wassim Kabalan
21373b89ee update code 2024-12-06 18:56:24 +01:00
Wassim KABALAN
f3b431aa74 add a user facing function to create uniform particle grid 2024-10-30 01:55:47 +01:00
Wassim KABALAN
a757b62f4b get local shape and zeros can be used by users 2024-10-30 01:55:20 +01:00
Wassim KABALAN
19011d0712 apply formating 2024-10-27 03:50:34 +01:00
Wassim KABALAN
cc4f310508 add distributed zeros 2024-10-27 03:48:17 +01:00
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