JaxPM/jaxpm
Wassim KABALAN 7d76573701
Some checks failed
Tests / run_tests (3.10) (push) Failing after 2m35s
Tests / run_tests (3.11) (push) Failing after 2m37s
Code Formatting / formatting (push) Successful in 3m1s
Tests / run_tests (3.12) (push) Failing after 1m16s
Replace np.power 0.5 by np.sqrt (#43)
* Switch **2  by np.sqrt

* format

---------

Co-authored-by: Francois Lanusse <fr.eiffel@gmail.com>
2025-06-29 10:51:45 +02:00
..
__init__.py Adds a trivial jaxpm implementation 2022-02-13 21:36:03 +01:00
distributed.py Fix sharding error (#37) 2025-06-28 23:07:31 +02:00
growth.py Replace np.power 0.5 by np.sqrt (#43) 2025-06-29 10:51:45 +02:00
kernels.py jaxdecomp proto (#21) 2024-12-20 05:44:02 -05: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 Fix sharding error (#37) 2025-06-28 23:07:31 +02:00
painting_utils.py Fix sharding error (#37) 2025-06-28 23:07:31 +02:00
plotting.py jaxdecomp proto (#21) 2024-12-20 05:44:02 -05:00
pm.py Replace np.power 0.5 by np.sqrt (#43) 2025-06-29 10:51:45 +02:00
utils.py Replace np.power 0.5 by np.sqrt (#43) 2025-06-29 10:51:45 +02:00