|
__init__.py
|
Adds a trivial jaxpm implementation
|
2022-02-13 21:36:03 +01:00 |
|
distributed.py
|
update distributed ops for new jd
|
2025-01-18 01:12:53 +01:00 |
|
growth.py
|
jaxdecomp proto (#21)
|
2024-12-20 05:44:02 -05:00 |
|
kernels.py
|
kernels now accept pytrees
|
2025-01-18 01:13:39 +01: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 painting functions to accept pytrees
|
2025-01-18 01:13:24 +01:00 |
|
plotting.py
|
jaxdecomp proto (#21)
|
2024-12-20 05:44:02 -05:00 |
|
pm.py
|
pm now accept pytrees
|
2025-01-18 01:13:51 +01:00 |
|
utils.py
|
jaxdecomp proto (#21)
|
2024-12-20 05:44:02 -05:00 |
