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

  • ab86699c88 merge with JZ Wassim KABALAN 2024-10-18 14:59:40 -04:00
  • 2ea05a1cd6 Update for jaxDecomp pure JAX Wassim KABALAN 2024-08-07 23:52:13 +02:00
  • ef7a7ef5c9 Merge pull request #23 from hsimonfroy/main Wassim KABALAN 2024-08-05 19:38:35 +02:00
  • ebe89e2ec6
         Merge pull request #23 from hsimonfroy/main Wassim KABALAN 2024-08-05 19:38:35 +02:00
  • 30060e82ea roll back painting Wassim KABALAN 2024-08-05 19:37:33 +02:00
  • 735699de77 roll back painting Wassim KABALAN 2024-08-05 19:37:33 +02:00
  • 8da3149581 fix formatting from main Wassim KABALAN 2024-08-03 00:45:35 +02:00
  • 02754cf452 added github workflow Wassim KABALAN 2024-08-03 00:44:05 +02:00
  • 783a97423e Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto Wassim KABALAN 2024-08-03 00:23:54 +02:00
  • ece8c93540 format Wassim KABALAN 2024-08-03 00:23:40 +02:00
  • 831291c1f9 bench Wassim KABALAN 2024-08-02 23:39:09 +02:00
  • 9af4659c81 add PMWD test Wassim KABALAN 2024-08-02 21:22:56 +02:00
  • aebc3e72c0 adjust test for hpc-plotter Wassim KABALAN 2024-08-02 21:22:45 +02:00
  • ccbfee3615 add assert to make pyright happy Wassim KABALAN 2024-08-02 21:21:59 +02:00
  • 75604d2396 Shared operation in fourrier space now take inverted sharding axis for slabs Wassim KABALAN 2024-08-02 21:21:00 +02:00
  • 8c5bd76c33 fix painting issue with slabs Wassim KABALAN 2024-08-02 21:20:15 +02:00
  • 9b21eed3b5 2lpt, get_ode, invlaplace, docstrings Hugo Simonfroy 2024-07-31 00:46:53 +02:00
  • d85867e88d 2lpt, get_ode, invlaplace, docstrings Hugo Simonfroy 2024-07-31 00:46:53 +02:00
  • f25eb7d465 Adjust pencil type for frequencies Wassim KABALAN 2024-07-28 13:44:03 +02:00
  • 1f2035176f
         Update scripts/distributed_pm.py Wassim KABALAN 2024-07-19 19:25:26 +02:00
  • b0b793e766 Merge pull request #22 from DifferentiableUniverseInitiative/all-contributors/add-dlanzieri Francois Lanusse 2024-07-19 10:52:37 -04:00
  • 2bb0f9e9a9
         Merge pull request #22 from DifferentiableUniverseInitiative/all-contributors/add-dlanzieri Francois Lanusse 2024-07-19 10:52:37 -04:00
  • 0d3bdda3ab docs: update .all-contributorsrc [skip ci] allcontributors[bot] 2024-07-19 14:51:26 +00:00
  • b4adc119b6
         docs: update .all-contributorsrc [skip ci] allcontributors[bot] 2024-07-19 14:51:26 +00:00
  • ad2ce8d825 docs: update README.md [skip ci] allcontributors[bot] 2024-07-19 14:51:25 +00:00
  • 38270f3358
         docs: update README.md [skip ci] allcontributors[bot] 2024-07-19 14:51:25 +00:00
  • 0f81a89d57 Merge pull request #14 from DifferentiableUniverseInitiative/neural_ode Francois Lanusse 2024-07-19 10:50:36 -04:00
  • d2ab0fe6e9
         Merge pull request #14 from DifferentiableUniverseInitiative/neural_ode Francois Lanusse 2024-07-19 10:50:36 -04:00
  • b949827e92 Update jaxpm/pm.py Francois Lanusse 2024-07-19 10:49:51 -04:00
  • 9cbdf18932
         Update jaxpm/pm.py Francois Lanusse 2024-07-19 10:49:51 -04:00
  • 9a279d2d6c Merge branch 'main' into neural_ode Francois Lanusse 2024-07-19 10:48:09 -04:00
  • 945dae4f80
         Merge branch 'main' into neural_ode Francois Lanusse 2024-07-19 10:48:09 -04:00
  • 5b7f595c32 Visualize only final field Wassim KABALAN 2024-07-18 17:05:22 +02:00
  • 0216837033 update benchmark and add slurm Wassim KABALAN 2024-07-18 17:04:52 +02:00
  • ed8cf8e532 add lpt2 Wassim KABALAN 2024-07-18 17:04:05 +02:00
  • 1f6b9c3217 times in ms Wassim KABALAN 2024-07-18 13:23:53 +02:00
  • 5f6d42eaeb add optimised bench script Wassim KABALAN 2024-07-18 13:16:37 +02:00
  • 4f508b7cb6 format Wassim KABALAN 2024-07-18 13:15:39 +02:00
  • 82be56836a add example Wassim KABALAN 2024-07-18 12:45:56 +02:00
  • abde5439f6 Return normal order frequencies for single GPU Wassim KABALAN 2024-07-18 12:44:18 +02:00
  • c81d4d2336 Wrap interpolation function to avoid all gather Wassim KABALAN 2024-07-18 12:43:52 +02:00
  • 7f48cfa8af Fix seed for distributed normal Wassim KABALAN 2024-07-18 12:43:24 +02:00
  • 7501b5bc6d Use new cic_paint with halo Wassim KABALAN 2024-07-18 12:42:07 +02:00
  • 5775a37550 implement distributed optimized cic_paint Wassim KABALAN 2024-07-18 12:39:15 +02:00
  • e62cd84cbd apply formatting Wassim KABALAN 2024-07-18 12:38:42 +02:00
  • ac86468c7c add halo exchange and slice pad Wassim KABALAN 2024-07-10 00:25:52 +02:00
  • 319942a6bc update formatting EiffL 2024-07-09 18:02:57 -04:00
  • 6408aff1de put back old functgion EiffL 2024-07-09 18:00:35 -04:00
  • ac9fdde948
         Merge branch 'main' into distributed_pm Francois Lanusse 2024-07-09 17:47:30 -04:00
  • a742065ffd adding example of distributed solution EiffL 2024-07-09 17:45:28 -04:00
  • f28442bb48 Applying formatting EiffL 2024-07-09 14:54:34 -04:00
  • a2811c0606 Applying formatting EiffL 2024-07-09 14:54:34 -04:00
  • be9a496161 Add draft example Wassim KABALAN 2024-07-09 02:35:53 +02:00
  • 1cade569af fixes for SPMD config Wassim KABALAN 2024-07-09 02:35:37 +02:00
  • f3599e73da Fixes in solvers Wassim KABALAN 2024-07-09 02:35:24 +02:00
  • a0be772f3c Add transpose for single GPU Wassim KABALAN 2024-07-09 02:35:13 +02:00
  • d2fb1ee1e2 Adjust painting operators Wassim KABALAN 2024-07-09 02:34:49 +02:00
  • 6d8f130be7 Import at the start to register everything Wassim KABALAN 2024-07-09 02:34:40 +02:00
  • e506dc2a7e add interpolate operator Wassim KABALAN 2024-07-09 02:34:24 +02:00
  • c99b2c3ef4 remove fftk from kernels.py Wassim KABALAN 2024-07-08 07:04:27 +02:00
  • da06f4dba8 Add solvers Wassim KABALAN 2024-07-08 00:24:07 +02:00
  • 8b9287184a Export operators Wassim KABALAN 2024-07-08 00:23:53 +02:00
  • bc2612a198 Implement Ops Wassim KABALAN 2024-07-08 00:23:35 +02:00
  • e708f5b176 Create operator infra Wassim KABALAN 2024-07-08 00:23:12 +02:00
  • 18d0bec2b5 add to gitignore Wassim KABALAN 2024-07-08 00:22:44 +02:00
  • 5b443e95c5 run precommit Wassim KABALAN 2024-07-08 00:22:29 +02:00
  • 38e875a7df add precommit Wassim KABALAN 2024-07-08 00:21:33 +02:00
  • f49ed6fe70 Add design Wassim KABALAN 2024-07-05 20:31:57 +02:00
  • febef01204 add plots Wassim KABALAN 2024-04-19 11:19:55 +02:00
  • 179030377b pm ok Wassim KABALAN 2024-04-19 10:32:38 +02:00
  • 055ceedb7e temp commit Wassim KABALAN 2024-04-19 01:11:25 +02:00
  • bf5f6e2d40
         Merge a67fc42495 into 5f463450d1 Daniel Forero 2023-05-18 16:31:28 -05:00
  • 835fa89aec Adding option to have weights in the 3d cic paint Francois Lanusse 2023-05-05 19:00:08 +02:00
  • 5f463450d1
         Adding option to have weights in the 3d cic paint Francois Lanusse 2023-05-05 19:00:08 +02:00
  • 6ca4c9191e Adding an example of jaxdecomp implementation EiffL 2022-11-26 17:27:14 +01:00
  • 6644b35d71 added fixes to alltoall EiffL 2022-11-05 23:25:01 +01:00
  • 72ae0fd88f fixed a whole lot of issues EiffL 2022-10-22 15:58:32 -04:00
  • 429813ad92 Implemented a few fixes to the FFT EiffL 2022-10-22 13:23:13 -04:00
  • 1948eae9ed Adding begnning of implem EiffL 2022-10-22 11:30:25 -05:00
  • 3c1abbafcd adds mpi tools EiffL 2022-10-22 08:30:43 -04:00
  • d4673d2955 Merge pull request #16 from DifferentiableUniverseInitiative/u/EiffL/distributed Francois Lanusse 2022-10-22 07:23:51 -04:00
  • 2db7c43ab8
         Merge pull request #16 from DifferentiableUniverseInitiative/u/EiffL/distributed Francois Lanusse 2022-10-22 07:23:51 -04:00
  • ada6ff9b70 removed old files EiffL 2022-10-22 07:17:44 -04:00
  • d9578b1220 removed old files EiffL 2022-10-22 07:17:44 -04:00
  • aa90804536 moved xmap stuff to experimental EiffL 2022-10-22 07:17:29 -04:00
  • a6096ae04b moved xmap stuff to experimental EiffL 2022-10-22 07:17:29 -04:00
  • 70600fb7f7 added things EiffL 2022-10-20 21:22:39 -07:00
  • a2e8017bc8 added things EiffL 2022-10-20 21:22:39 -07:00
  • ebab27b4c3 update data EiffL 2022-10-20 21:03:15 -07:00
  • 953fc48cc0 update data EiffL 2022-10-20 21:03:15 -07:00
  • 137f4e5099 adding distributed ops EiffL 2022-10-20 23:05:39 -04:00
  • 7f3de80f7a adding distributed ops EiffL 2022-10-20 23:05:39 -04:00
  • 1cb2739366 Added prototype distributed LPT EiffL 2022-10-19 17:37:38 -07:00
  • 2da21d8b5e Added prototype distributed LPT EiffL 2022-10-19 17:37:38 -07:00
  • 64894726e7 Added example code to compute distributed fft EiffL 2022-10-17 22:07:12 -07:00
  • 61578217fe Added example code to compute distributed fft EiffL 2022-10-17 22:07:12 -07:00
  • a67fc42495 Adding fixamp ics Daniel Forero-Sanchez 2022-09-07 21:45:00 +02:00
  • 84b79af7f8 creoss correlation function denise lanzieri 2022-06-18 18:23:46 +02:00
  • b26cfcd01a creoss correlation function denise lanzieri 2022-06-18 18:23:46 +02:00
  • 8b885450a8 few adjustments to PGD correction denise lanzieri 2022-06-13 17:17:19 +02:00