JaxPM/tests
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
..
conftest.py Replace np.power 0.5 by np.sqrt (#43) 2025-06-29 10:51:45 +02:00
helpers.py jaxdecomp proto (#21) 2024-12-20 05:44:02 -05:00
test_against_fpm.py Fix sharding error (#37) 2025-06-28 23:07:31 +02:00
test_distributed_pm.py Fix sharding error (#37) 2025-06-28 23:07:31 +02:00
test_gradients.py Fix sharding error (#37) 2025-06-28 23:07:31 +02:00