JaxPM/jaxpm
Wassim KABALAN ed8cf8e532 add lpt2
2024-07-18 17:04:05 +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 format 2024-07-18 13:16:14 +02:00
growth.py Applying formatting 2024-07-09 14:54:34 -04:00
kernels.py Return normal order frequencies for single GPU 2024-07-18 12:44:18 +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 implement distributed optimized cic_paint 2024-07-18 12:39:15 +02:00
painting_utils.py implement distributed optimized cic_paint 2024-07-18 12:39:15 +02:00
pm.py add lpt2 2024-07-18 17:04:05 +02:00
utils.py Applying formatting 2024-07-09 14:54:34 -04:00