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
JaxPM/jaxpm
History
Wassim KABALAN 2ea05a1cd6 Update for jaxDecomp pure JAX
2024-10-18 20:45:08 +02:00
..
experimental Applying formatting 2024-07-09 14:54:34 -04:00
__init__.py Adds a trivial jaxpm implementation 2022-02-13 21:36:03 +01:00
distributed.py Update for jaxDecomp pure JAX 2024-10-18 20:45:08 +02:00
growth.py Applying formatting 2024-07-09 14:54:34 -04:00
kernels.py Shared operation in fourrier space now take inverted sharding axis for 2024-08-02 21:21:06 +02:00
lensing.py Applying formatting 2024-07-09 14:54:34 -04:00
nn.py Applying formatting 2024-07-09 14:54:34 -04:00
painting.py Update for jaxDecomp pure JAX 2024-10-18 20:45:08 +02:00
painting_utils.py implement distributed optimized cic_paint 2024-07-18 12:39:15 +02:00
pm.py add assert to make pyright happy 2024-08-02 21:21:59 +02:00
utils.py Applying formatting 2024-07-09 14:54:34 -04:00