guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-28 08:01:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Commit graph

  • 81dcf3ef10
         Merge e666aada42 into cb2a7ab17f Wassim KABALAN 2025-06-28 00:57:23 +00:00
  • e666aada42
         Update conftest.py fix-sharding-error Francois Lanusse 2025-06-28 02:57:20 +02:00
  • 844cace3d7 adding caching Francois Lanusse 2025-06-28 02:41:16 +02:00
  • 37976c70d8 fixed tolerance Francois Lanusse 2025-06-28 02:36:37 +02:00
  • 0c75158043 fix accuracy Francois Lanusse 2025-06-28 02:21:19 +02:00
  • 627be7a764 updating tests Francois Lanusse 2025-06-28 01:36:17 +02:00
  • 7623e60581 debug test workflow Wassim Kabalan 2025-06-12 15:28:03 +02:00
  • d874790543 format Wassim Kabalan 2025-06-12 15:20:20 +02:00
  • 187cf5c4ba debug test workflow Wassim Kabalan 2025-06-12 15:19:36 +02:00
  • 67a80e1041 format Wassim Kabalan 2025-06-12 14:51:24 +02:00
  • e1a8134b8e update by default normal_field dtype to match JAX Wassim Kabalan 2025-06-12 14:46:27 +02:00
  • 7834305478
         Merge cf979f3250 into cb2a7ab17f pointeee 2025-06-10 16:00:41 +00:00
  • cf979f3250 ? chenze@igpu 2025-06-10 22:46:26 +09:00
  • f391a79211 make normal_field signature consistent with jax.random.normal Wassim Kabalan 2025-06-09 19:35:40 +02:00
  • 5807e1d3f4 format Wassim Kabalan 2025-06-08 15:37:06 +02:00
  • a7fcba0e1f update wf Wassim Kabalan 2025-06-08 15:35:57 +02:00
  • c6a7dd4e4e update tests Wassim Kabalan 2025-06-08 15:27:32 +02:00
  • e0ba85fb58 update tests.yml Wassim Kabalan 2025-06-08 12:03:44 +02:00
  • 12eddc4e6a add mpi4py Wassim Kabalan 2025-06-08 11:48:11 +02:00
  • 0eb4c371e3 update tests Wassim Kabalan 2025-06-08 11:43:15 +02:00
  • 3be619a2db reorganize install in test workflow Wassim Kabalan 2025-06-08 11:36:06 +02:00
  • f0b849cf5f update tolerance :) Wassim Kabalan 2025-06-08 11:35:57 +02:00
  • 995cc4c78c update numpy install in wf Wassim Kabalan 2025-06-08 11:26:32 +02:00
  • 7b7205e3b3 update notebooks/03-MultiGPU_PM_Halo.ipynb Wassim Kabalan 2025-06-08 11:25:51 +02:00
  • 6aacd81bd6 update test jax version Wassim Kabalan 2025-06-08 10:56:34 +02:00
  • 49c93aacf6 Format Wassim Kabalan 2025-06-08 10:45:20 +02:00
  • 41ae41ace3 Allow env variable control of caching in growth Wassim Kabalan 2025-06-08 10:45:04 +02:00
  • e7112e0c25 update cic read halo size and notebooks examples Wassim Kabalan 2025-06-07 19:26:37 +02:00
  • d4049e5db4 fix 2D painting when input is (X , Y , 2) shape Wassim Kabalan 2025-05-09 22:07:53 +02:00
  • 20ace41d32 update Growth.py to allow using FastPM solver Wassim Kabalan 2025-05-08 15:08:28 +02:00
  • e1daa8cba4 final touches Wassim Kabalan 2025-02-28 14:03:33 +01:00
  • 4e4d3745f0 add Caveats to notebook readme Wassim Kabalan 2025-02-28 13:48:08 +01:00
  • 9f494da317 format Wassim Kabalan 2025-02-28 13:47:43 +01:00
  • 580387ce1c update distributed tests and add jacfwd jacrev and vmap tests Wassim Kabalan 2025-02-28 13:47:22 +01:00
  • eb0778aa24 update test_against_fpm Wassim Kabalan 2025-02-28 13:47:00 +01:00
  • 91d3292923 add correct annotations for weights in painting and warning for cic_paint in distributed pm Wassim Kabalan 2025-02-28 13:46:41 +01:00
  • f8325b1c67 fix tests Wassim Kabalan 2025-02-28 10:06:09 +01:00
  • f3b8f4160e notebook update Wassim Kabalan 2025-02-28 09:59:04 +01:00
  • b43cb373a0 format Wassim Kabalan 2025-02-28 09:56:00 +01:00
  • 51ee4dd937 update examples Wassim Kabalan 2025-02-28 09:54:53 +01:00
  • 8e0f300572 Update examples Wassim Kabalan 2025-02-26 14:32:47 +01:00
  • 0b08c6f59a Use cosmo as arg for the ODE function Wassim Kabalan 2025-02-26 14:05:25 +01:00
  • 9b8e42aa00 updates for FullField shardedmap-jaxdecomp Wassim Kabalan 2025-02-05 14:26:28 +01:00
  • 34bf577b1b small fix Wassim Kabalan 2025-01-21 12:08:47 +01:00
  • d4f72c8244 format Wassim Kabalan 2025-01-21 11:21:35 +01:00
  • 6aa2ccf7b5 don't send static cosmo into ODETerm Wassim Kabalan 2025-01-21 11:18:00 +01:00
  • 4ac82abe39 remove extra text Wassim Kabalan 2025-01-21 10:15:11 +01:00
  • 6ab26ea1ec remove extra text Wassim Kabalan 2025-01-21 10:15:11 +01:00
  • 0b723c21ae bump minimum jd version Wassim Kabalan 2025-01-20 23:08:12 +01:00
  • 03610602a8 bump minimum jd version Wassim Kabalan 2025-01-20 23:08:12 +01:00
  • a8f11a75ea format Wassim Kabalan 2025-01-20 22:47:07 +01:00
  • bf12da3bb7 Merge remote-tracking branch 'origin/main' into fix-pfft-gradients Wassim Kabalan 2025-01-20 22:45:32 +01:00
  • 1f5c619531 format Wassim Kabalan 2025-01-20 22:41:19 +01:00
  • 20fe25c324 update tests and add test for FWD REV gradient Wassim Kabalan 2025-01-20 22:40:28 +01:00
  • 151fa09247 power spec should accept pytrees Wassim Kabalan 2025-01-20 22:40:01 +01:00
  • 38f6599974 painting now accepts pytrees Wassim Kabalan 2025-01-20 22:39:40 +01:00
  • 9e203b5680 pm now accept pytrees Wassim Kabalan 2025-01-18 01:13:51 +01:00
  • 204a9526ec kernels now accept pytrees Wassim Kabalan 2025-01-18 01:13:39 +01:00
  • f5755b4b5d update painting functions to accept pytrees Wassim Kabalan 2025-01-18 01:13:24 +01:00
  • 7c3577ea71 update distributed ops for new jd Wassim Kabalan 2025-01-18 01:12:53 +01:00
  • cb2a7ab17f
         Fix pfft gradients (#34) main 38-replace-nppower-by-npsqrt Wassim KABALAN 2024-12-22 18:47:42 +01:00
  • bbacd45dcf format Wassim Kabalan 2024-12-21 23:27:05 +01:00
  • a924458f0d Prepare for DTO tests Wassim Kabalan 2024-12-21 23:24:19 +01:00
  • b132a0e2aa update jaxdecomp version and test gradients Wassim Kabalan 2024-12-21 23:14:45 +01:00
  • 45aa70899a adding more information in contributing additional_testing Francois Lanusse 2024-12-21 16:29:51 -05:00
  • e316d8a0b2 adding caching of dependencies Francois Lanusse 2024-12-21 15:57:40 -05:00
  • d81a2529e7 minor typo fix Francois Lanusse 2024-12-21 15:28:20 -05:00
  • 15f2fb1ee6 adding notice Francois Lanusse 2024-12-21 15:26:53 -05:00
  • ae0f439ae4 fixing formatting of notebook Francois Lanusse 2024-12-21 13:14:42 -05:00
  • ea9fbf6aa8
         Update README.md Francois Lanusse 2024-12-21 13:13:37 -05:00
  • ad16a0659a Created using Colab Francois Lanusse 2024-12-21 13:10:15 -05:00
  • f245a1f685
         Pypi upload compatible version (#33) v0.1.5 notebook_updates Francois Lanusse 2024-12-21 11:47:13 -05:00
  • 160b86eb71
         Create python-publish.yml Francois Lanusse 2024-12-21 11:38:24 -05:00
  • def149499a Adding very simple install instructions pypi_version Francois Lanusse 2024-12-21 11:32:58 -05:00
  • 953c36f582 Adding badge for pypi version Francois Lanusse 2024-12-21 11:30:21 -05:00
  • c888fa3bee Fixing formatting Francois Lanusse 2024-12-21 11:28:36 -05:00
  • a2da779026 updating name of project Francois Lanusse 2024-12-21 11:16:13 -05:00
  • e7443a53a3 adding manifest to remove unecessary files Francois Lanusse 2024-12-21 11:12:50 -05:00
  • 7540ad1d3f moving test dependencies separately Francois Lanusse 2024-12-21 11:04:26 -05:00
  • f14f0fe68e
         Changing description (#32) v0.1.1 Francois Lanusse 2024-12-21 10:45:12 -05:00
  • 1a9b67a371 Changing description pypi Francois Lanusse 2024-12-21 10:44:00 -05:00
  • 70ab9f1931 remove deprecated stuff Francois Lanusse 2024-12-21 09:58:42 -05:00
  • bc6e57532d remove massive file Francois Lanusse 2024-12-21 09:58:18 -05:00
  • 4b4450d7d3 Remove huge notebook from history (#30) Wassim KABALAN 2024-12-20 14:43:43 +01:00
  • d3e6d55238
         Update pyproject.toml Francois Lanusse 2024-12-20 08:43:31 -05:00
  • 96e0126b08
         Merge branch 'main' into ASKabalan/patch Francois Lanusse 2024-12-20 08:41:05 -05:00
  • dd161f7364 Adding version (#29) v0.1.0 Francois Lanusse 2024-12-20 07:45:29 -05:00
  • 134aa4cbeb
         Adding version (#29) Francois Lanusse 2024-12-20 07:45:29 -05:00
  • 6cf7daa4a3 Deleting Animating notebook Wassim Kabalan 2024-12-20 13:35:24 +01:00
  • 832f8cb71c Update README.md Francois Lanusse 2024-12-20 05:56:16 -05:00
  • 17623c5081
         Update README.md Francois Lanusse 2024-12-20 05:56:16 -05:00
  • fe07df0cd8 Update README.md Francois Lanusse 2024-12-20 05:56:16 -05:00
  • 651bddf832
         fixing version file Francois Lanusse 2024-12-20 05:54:51 -05:00
  • 281f87ab6a
         adding version Francois Lanusse 2024-12-20 05:54:17 -05:00
  • df8602b318 jaxdecomp proto (#21) Wassim KABALAN 2024-12-20 11:44:02 +01:00
  • 695e359f56 jaxdecomp proto (#21) Wassim KABALAN 2024-12-20 11:44:02 +01:00
  • bf44dfdea9
         jaxdecomp proto (#21) Wassim KABALAN 2024-12-20 11:44:02 +01:00
  • f91aa931dc remove outdated tests Wassim Kabalan 2024-12-09 10:18:48 +01:00
  • 158478cb4a revert test workflow Wassim Kabalan 2024-12-09 10:06:16 +01:00
  • 3be5daeda5 bump jaxdecomp version Wassim Kabalan 2024-12-09 10:03:36 +01:00