JaxPM/jaxpm
2024-10-22 17:30:33 -04: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 quick fix in distributed 2024-10-22 12:15:37 -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 remove duplicate get_ode_fn 2024-10-22 13:38:16 -04:00
utils.py fix formatting from main 2024-08-03 00:45:35 +02:00