guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-28 16:11:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
145 commits 9 branches 5 tags 172 MiB
Commit graph

145 commits

This branch
This branch
All branches
Author SHA1 Message Date
Francois Lanusse
e666aada42
Update conftest.py 2025-06-28 02:57:20 +02:00
Francois Lanusse
844cace3d7 adding caching 2025-06-28 02:41:16 +02:00
Francois Lanusse
37976c70d8 fixed tolerance 2025-06-28 02:36:37 +02:00
Francois Lanusse
0c75158043 fix accuracy 2025-06-28 02:21:19 +02:00
Francois Lanusse
627be7a764 updating tests 2025-06-28 01:36:17 +02:00
Wassim Kabalan
7623e60581 debug test workflow 2025-06-12 15:28:03 +02:00
Wassim Kabalan
d874790543 format 2025-06-12 15:20:20 +02:00
Wassim Kabalan
187cf5c4ba debug test workflow 2025-06-12 15:19:36 +02:00
Wassim Kabalan
67a80e1041 format 2025-06-12 14:51:24 +02:00
Wassim Kabalan
e1a8134b8e update by default normal_field dtype to match JAX 2025-06-12 14:46:27 +02:00
Wassim Kabalan
f391a79211 make normal_field signature consistent with jax.random.normal 2025-06-09 19:35:40 +02:00
Wassim Kabalan
5807e1d3f4 format 2025-06-08 15:37:06 +02:00
Wassim Kabalan
a7fcba0e1f update wf 2025-06-08 15:35:57 +02:00
Wassim Kabalan
c6a7dd4e4e update tests 2025-06-08 15:27:32 +02:00
Wassim Kabalan
e0ba85fb58 update tests.yml 2025-06-08 12:03:44 +02:00
Wassim Kabalan
12eddc4e6a add mpi4py 2025-06-08 11:48:11 +02:00
Wassim Kabalan
0eb4c371e3 update tests 2025-06-08 11:43:15 +02:00
Wassim Kabalan
3be619a2db reorganize install in test workflow 2025-06-08 11:36:06 +02:00
Wassim Kabalan
f0b849cf5f update tolerance :) 2025-06-08 11:35:57 +02:00
Wassim Kabalan
995cc4c78c update numpy install in wf 2025-06-08 11:26:32 +02:00
Wassim Kabalan
7b7205e3b3 update notebooks/03-MultiGPU_PM_Halo.ipynb 2025-06-08 11:25:51 +02:00
Wassim Kabalan
6aacd81bd6 update test jax version 2025-06-08 10:56:34 +02:00
Wassim Kabalan
49c93aacf6 Format 2025-06-08 10:45:20 +02:00
Wassim Kabalan
41ae41ace3 Allow env variable control of caching in growth 2025-06-08 10:45:04 +02:00
Wassim Kabalan
e7112e0c25 update cic read halo size and notebooks examples 2025-06-07 19:26:37 +02:00
Wassim Kabalan
d4049e5db4 fix 2D painting when input is (X , Y , 2) shape 2025-05-09 22:07:53 +02:00
Wassim Kabalan
20ace41d32 update Growth.py to allow using FastPM solver 2025-05-08 15:08:28 +02:00
Wassim Kabalan
e1daa8cba4 final touches 2025-02-28 14:03:33 +01:00
Wassim Kabalan
4e4d3745f0 add Caveats to notebook readme 2025-02-28 13:48:08 +01:00
Wassim Kabalan
9f494da317 format 2025-02-28 13:47:43 +01:00
Wassim Kabalan
580387ce1c update distributed tests and add jacfwd jacrev and vmap tests 2025-02-28 13:47:22 +01:00
Wassim Kabalan
eb0778aa24 update test_against_fpm 2025-02-28 13:47:00 +01:00
Wassim Kabalan
91d3292923 add correct annotations for weights in painting and warning for cic_paint in distributed pm 2025-02-28 13:46:41 +01:00
Wassim Kabalan
f8325b1c67 fix tests 2025-02-28 10:06:09 +01:00
Wassim Kabalan
f3b8f4160e notebook update 2025-02-28 09:59:04 +01:00
Wassim Kabalan
b43cb373a0 format 2025-02-28 09:56:00 +01:00
Wassim Kabalan
51ee4dd937 update examples 2025-02-28 09:54:53 +01:00
Wassim Kabalan
8e0f300572 Update examples 2025-02-26 14:32:47 +01:00
Wassim Kabalan
0b08c6f59a Use cosmo as arg for the ODE function 2025-02-26 14:05:25 +01:00
Wassim KABALAN
cb2a7ab17f
Fix pfft gradients (#34) 
* update jaxdecomp version and test gradients

* Prepare for DTO tests

* format
 2024-12-22 12:47:42 -05:00
Francois Lanusse
d81a2529e7 minor typo fix 2024-12-21 15:28:20 -05:00
Francois Lanusse
15f2fb1ee6 adding notice 2024-12-21 15:26:53 -05:00
Francois Lanusse
ae0f439ae4 fixing formatting of notebook 2024-12-21 13:14:42 -05:00
Francois Lanusse
ea9fbf6aa8
Update README.md 2024-12-21 13:13:37 -05:00
Francois Lanusse
ad16a0659a Created using Colab 2024-12-21 13:10:15 -05:00
Francois Lanusse
f245a1f685
Pypi upload compatible version (#33) 
* moving test dependencies separately

* adding manifest to remove unecessary files

* updating name of project

* Fixing formatting

* Adding badge for pypi version

* Adding very simple install instructions
 2024-12-21 11:47:13 -05:00
Francois Lanusse
160b86eb71
Create python-publish.yml 2024-12-21 11:38:24 -05:00
Francois Lanusse
f14f0fe68e
Changing description (#32) 2024-12-21 10:45:12 -05:00
Francois Lanusse
70ab9f1931 remove deprecated stuff 
Former-commit-id: da6fdea0556407c968dd31d44af832c997f2645c
 2024-12-21 09:58:42 -05:00
Francois Lanusse
bc6e57532d remove massive file 
Former-commit-id: 69f49ba2ed23c16bcea70024ea3bcff4d71b8a5b
 2024-12-21 09:58:18 -05:00