guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-04-24 03:40:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Commit graph

  • 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
  • f185b8b913 few adjustments to PGD correction denise lanzieri 2022-06-13 17:17:19 +02:00
  • d8a1dbe210 neural ode added denise lanzieri 2022-06-11 14:28:30 +02:00
  • 5b72a73448 neural ode added denise lanzieri 2022-06-11 14:28:30 +02:00
  • 20fc4a5562 Merge pull request #13 from DifferentiableUniverseInitiative/u/EiffL/lensing Francois Lanusse 2022-05-18 15:55:40 +02:00
  • 101ef80d9c
         Merge pull request #13 from DifferentiableUniverseInitiative/u/EiffL/lensing Francois Lanusse 2022-05-18 15:55:40 +02:00
  • cd7a48e463 fixes the lensing demo EiffL 2022-05-18 15:43:45 +02:00
  • 5c4f75a102 fixes the lensing demo EiffL 2022-05-18 15:43:45 +02:00
  • 301bd37647 changes definition of lensplanes EiffL 2022-05-18 15:40:11 +02:00
  • 1b7c797286 changes definition of lensplanes EiffL 2022-05-18 15:40:11 +02:00
  • 0991789553 Merge pull request #11 from DifferentiableUniverseInitiative/u/EiffL/lensing v0.0.1 Francois Lanusse 2022-05-18 14:35:06 +02:00
  • e7fa6c9e57
         Merge pull request #11 from DifferentiableUniverseInitiative/u/EiffL/lensing Francois Lanusse 2022-05-18 14:35:06 +02:00
  • ff5fe8069e Update jaxpm/painting.py Francois Lanusse 2022-05-18 10:22:21 +02:00
  • 4a584b1d9c
         Update jaxpm/painting.py Francois Lanusse 2022-05-18 10:22:21 +02:00
  • e93aa07f93 Merge pull request #12 from DifferentiableUniverseInitiative/PGD Francois Lanusse 2022-05-18 10:21:01 +02:00
  • 9967dffd82
         Merge pull request #12 from DifferentiableUniverseInitiative/PGD Francois Lanusse 2022-05-18 10:21:01 +02:00
  • 3df0c05d29 Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:59:59 +02:00
  • 5394abbed1
         Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:59:59 +02:00
  • dd71d359e0 Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:58:46 +02:00
  • 782c9dbc6c
         Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:58:46 +02:00
  • 77827fcf44 Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:58:34 +02:00
  • 8266f11369
         Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:58:34 +02:00
  • 85b2f4f097 Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:55:31 +02:00
  • b737631346
         Update jaxpm/pm.py Denise Lanzieri 2022-05-18 09:55:31 +02:00
  • 60309e1490 small fix EiffL 2022-05-17 23:49:12 +02:00
  • 0f6fb39966 small fix EiffL 2022-05-17 23:49:12 +02:00
  • aff8db56f5 change impor EiffL 2022-05-17 23:42:57 +02:00
  • a52287cd97 change impor EiffL 2022-05-17 23:42:57 +02:00
  • 5108e56ee8 adds fix to make code jittablel EiffL 2022-05-17 23:37:55 +02:00
  • 88cff99736 adds fix to make code jittablel EiffL 2022-05-17 23:37:55 +02:00