Wassim KABALAN 5d4f438e92 update PGDCorrection and neural ode to use new fft3d		2024-10-25 10:14:29 +02:00
..
__init__.py	Adds a trivial jaxpm implementation	2022-02-13 21:36:03 +01:00
distributed.py	By default check_rep is false for shard_map	2024-10-22 17:30:56 -04:00
growth.py	Applying formatting	2024-07-09 14:54:34 -04:00
kernels.py	update compensate CIC	2024-10-22 17:30:33 -04: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 compensate CIC	2024-10-22 17:30:33 -04:00
painting_utils.py	implement distributed optimized cic_paint	2024-07-18 12:39:15 +02:00
pm.py	update PGDCorrection and neural ode to use new fft3d	2024-10-25 10:14:29 +02:00
utils.py	fix formatting from main	2024-08-03 00:45:35 +02:00