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

11 commits

Author SHA1 Message Date
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
f25eb7d465 Adjust pencil type for frequencies 2024-07-28 13:44:03 +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
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
EiffL
a2811c0606 Applying formatting 2024-07-09 14:54:34 -04:00
denise lanzieri
d3026f7849 PGD 2022-05-17 15:28:30 +02:00
EiffL
d8a5b702a9 adding cic compensation tools 2022-03-26 00:06:34 +01:00
EiffL
afdfcfa644 Adds a trivial jaxpm implementation 2022-02-13 21:36:03 +01:00