guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-14 09:51:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
JaxPM/jaxpm
History
Wassim Kabalan f391a79211 make normal_field signature consistent with jax.random.normal
2025-06-09 19:35:40 +02:00
..
__init__.py Adds a trivial jaxpm implementation 2022-02-13 21:36:03 +01:00
distributed.py make normal_field signature consistent with jax.random.normal 2025-06-09 19:35:40 +02:00
growth.py Allow env variable control of caching in growth 2025-06-08 10:45:04 +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 Format 2025-06-08 10:45:20 +02:00
painting_utils.py format 2025-06-08 15:37:06 +02:00
plotting.py jaxdecomp proto (#21) 2024-12-20 05:44:02 -05:00
pm.py make normal_field signature consistent with jax.random.normal 2025-06-09 19:35:40 +02:00
utils.py jaxdecomp proto (#21) 2024-12-20 05:44:02 -05:00