guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-02-22 17:47:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Commit graph

  • ad4566620f clarifying the halo size dependence on the box size Wassim KABALAN 2024-11-10 20:58:29 +01:00
  • 2472a5d9df clarifying why cic_paint_dx is slower Wassim KABALAN 2024-11-10 20:58:03 +01:00
  • 6256fbadb1 forgot particles in multi host Wassim KABALAN 2024-10-30 02:30:41 +01:00
  • 42d8e89de4 fix small error Wassim KABALAN 2024-10-30 02:03:54 +01:00
  • a067954d0a update readme Wassim KABALAN 2024-10-30 01:59:33 +01:00
  • f0c43f8d76 add readme for tutorials Wassim KABALAN 2024-10-30 01:59:23 +01:00
  • a030ec47c6 delete outdated design file Wassim KABALAN 2024-10-30 01:59:07 +01:00
  • 72457d6c37 update example notebooks Wassim KABALAN 2024-10-30 01:58:37 +01:00
  • 4da4c66472 remove script folder Wassim KABALAN 2024-10-30 01:58:17 +01:00
  • e9529d35f8 Weights can be traced Wassim KABALAN 2024-10-30 01:57:56 +01:00
  • b09580d59e Allow applying weights with relative cic paint Wassim KABALAN 2024-10-30 01:57:32 +01:00
  • b3a264ad53 use float64 for enmeshing Wassim KABALAN 2024-10-30 01:56:29 +01:00
  • 2ad035a75b use jax interp instead of jax_cosmo Wassim KABALAN 2024-10-30 01:56:04 +01:00
  • f3b431aa74 add a user facing function to create uniform particle grid Wassim KABALAN 2024-10-30 01:55:47 +01:00
  • a757b62f4b get local shape and zeros can be used by users Wassim KABALAN 2024-10-30 01:55:20 +01:00
  • 19011d0712 apply formating Wassim KABALAN 2024-10-27 03:50:34 +01:00
  • c93894f561 update notebooks Wassim KABALAN 2024-10-27 03:49:07 +01:00
  • b4fdb74660 jit cic_paint Wassim KABALAN 2024-10-27 03:48:50 +01:00
  • d62c38f457 fix code in LPT2 Wassim KABALAN 2024-10-27 03:48:38 +01:00
  • cc4f310508 add distributed zeros Wassim KABALAN 2024-10-27 03:48:17 +01:00
  • 4342279817 apply formatting Wassim KABALAN 2024-10-27 00:52:14 +02:00
  • 11f7e90066 Merge remote-tracking branch 'upstream/ASKabalan/jaxdecomp_proto' into ASKabalan/jaxdecomp_proto Wassim KABALAN 2024-10-27 00:51:16 +02:00
  • 49dd18a3f8 add notebook examples Wassim KABALAN 2024-10-26 22:47:26 +02:00
  • 0c96a4dc10 delete old slurms Wassim KABALAN 2024-10-26 22:47:11 +02:00
  • ff8856d2bc use jnp interp instead of jc interp Wassim KABALAN 2024-10-26 18:53:11 +02:00
  • d2f1eb2fa4 fix painting issue with read_cic Wassim KABALAN 2024-10-26 18:52:51 +02:00
  • 0f833f0cb4 remove deprecated stuff Francois Lanusse 2024-10-24 16:36:41 -04:00
  • 69c35d1aa8 Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto Wassim KABALAN 2024-10-25 11:04:55 +02:00
  • 8e8e8964be jaxDecomp pfft3d promotes to complex automatically Wassim KABALAN 2024-10-25 10:15:43 +02:00
  • 5d4f438e92 update PGDCorrection and neural ode to use new fft3d Wassim KABALAN 2024-10-25 10:14:29 +02:00
  • 505f2ec286 remove experimental distributed code Wassim KABALAN 2024-10-25 10:05:33 +02:00
  • f57e32af7f remove deprecated stuff Francois Lanusse 2024-10-24 16:36:41 -04:00
  • a848d43b56
         remove deprecated stuff Francois Lanusse 2024-10-24 16:36:41 -04:00
  • 45b2c7f0d2 By default check_rep is false for shard_map Wassim KABALAN 2024-10-22 17:30:56 -04:00
  • 0bb992fc56 update compensate CIC Wassim KABALAN 2024-10-22 17:30:20 -04:00
  • cf799b6520 update visualizer Wassim KABALAN 2024-10-22 13:39:03 -04:00
  • 31ca41b0a7 remove duplicate get_ode_fn Wassim KABALAN 2024-10-22 13:38:16 -04:00
  • 82f29877f5 format Wassim KABALAN 2024-10-22 13:24:24 -04:00
  • d28982eec7 Small fix Wassim KABALAN 2024-10-22 13:00:25 -04:00
  • 85cca44fb0 Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto Wassim KABALAN 2024-10-22 12:59:50 -04:00
  • 82b8f563a0 format Wassim KABALAN 2024-10-22 12:57:34 -04:00
  • 86233081e2 merge hugos LPT2 code Wassim KABALAN 2024-10-22 12:57:05 -04:00
  • 2f509932f5 update demo Wassim KABALAN 2024-10-22 12:16:28 -04:00
  • 0433c615f3 quick fix in distributed Wassim KABALAN 2024-10-22 12:15:37 -04:00
  • a8b194f326 quick fix in kernels Wassim KABALAN 2024-10-22 12:15:17 -04:00
  • 4d944f01f2 update demo script Wassim KABALAN 2024-10-22 12:10:18 -04:00
  • 105568e8db painting.py no longer uses global mesh Wassim KABALAN 2024-10-22 12:10:01 -04:00
  • 80c56dced5 pm.py no longer uses global mesh Wassim KABALAN 2024-10-22 12:09:50 -04:00
  • 56ffd263f6 quick fix in distributed Wassim KABALAN 2024-10-22 12:07:06 -04:00
  • a5b267bd63 kernels.py no longer uses global mesh Wassim KABALAN 2024-10-22 11:05:21 -04:00
  • 591ee32c55 global mesh no longer needed Wassim KABALAN 2024-10-22 11:04:29 -04:00
  • 38714cf65d add a strict dependency on jaxdecomp Wassim KABALAN 2024-10-22 11:02:55 -04:00
  • a160a3faa9 update to leapfrog Wassim KABALAN 2024-10-22 09:03:41 -04:00
  • 5a587fd402 merge bench_pm Wassim KABALAN 2024-10-21 14:04:09 -04:00
  • 375f2048e4 make distributed pm work in single controller Wassim KABALAN 2024-10-21 14:03:48 -04:00
  • 9c94f994ff format Wassim KABALAN 2024-10-21 14:00:14 -04:00
  • 0ce7219ad8 make normal_field work with single controller Wassim KABALAN 2024-10-21 13:59:00 -04:00
  • ff1c5e8362 remove fourrier_space in autoshmap Wassim KABALAN 2024-10-21 13:57:26 -04:00
  • 01b952701e update for latest jaxDecomp Wassim KABALAN 2024-10-21 13:55:48 -04:00
  • afecb13cde revert single halo extent change Wassim KABALAN 2024-10-20 09:58:24 -04:00
  • 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