Commit graph

101 commits

Author SHA1 Message Date
Wassim Kabalan
21373b89ee update code 2024-12-06 18:56:24 +01:00
Wassim Kabalan
e0c118a540 By default use absoulute painting with 2024-12-05 18:23:18 +01:00
Wassim Kabalan
c1b276d224 Put plotting utils in package 2024-12-05 18:22:34 +01:00
Wassim Kabalan
b32014b7ea Add finite difference laplace kernel + powerspec functions from Hugo
Co-authored-by: Hugo Simonfroy <hugo.simonfroy@gmail.com>
2024-12-05 18:21:24 +01:00
Wassim KABALAN
435c7c848f Put plotting in package 2024-11-14 13:52:59 +01:00
Wassim KABALAN
e9529d35f8 Weights can be traced 2024-10-30 01:57:56 +01:00
Wassim KABALAN
b09580d59e Allow applying weights with relative cic paint 2024-10-30 01:57:32 +01:00
Wassim KABALAN
b3a264ad53 use float64 for enmeshing 2024-10-30 01:56:29 +01:00
Wassim KABALAN
2ad035a75b use jax interp instead of jax_cosmo 2024-10-30 01:56:04 +01:00
Wassim KABALAN
f3b431aa74 add a user facing function to create uniform particle grid 2024-10-30 01:55:47 +01:00
Wassim KABALAN
a757b62f4b get local shape and zeros can be used by users 2024-10-30 01:55:20 +01:00
Wassim KABALAN
19011d0712 apply formating 2024-10-27 03:50:34 +01:00
Wassim KABALAN
b4fdb74660 jit cic_paint 2024-10-27 03:48:50 +01:00
Wassim KABALAN
d62c38f457 fix code in LPT2 2024-10-27 03:48:38 +01:00
Wassim KABALAN
cc4f310508 add distributed zeros 2024-10-27 03:48:17 +01:00
Wassim KABALAN
4342279817 apply formatting 2024-10-27 00:52:14 +02:00
Wassim KABALAN
ff8856d2bc use jnp interp instead of jc interp 2024-10-26 22:48:38 +02:00
Wassim KABALAN
d2f1eb2fa4 fix painting issue with read_cic 2024-10-26 22:48:38 +02:00
Francois Lanusse
0f833f0cb4 remove deprecated stuff 2024-10-26 22:48:38 +02:00
Wassim KABALAN
8e8e8964be jaxDecomp pfft3d promotes to complex automatically 2024-10-25 10:15:43 +02:00
Wassim KABALAN
5d4f438e92 update PGDCorrection and neural ode to use new fft3d 2024-10-25 10:14:29 +02:00
Wassim KABALAN
505f2ec286 remove experimental distributed code 2024-10-25 10:05:33 +02:00
Wassim KABALAN
45b2c7f0d2 By default check_rep is false for shard_map 2024-10-22 17:30:56 -04:00
Wassim KABALAN
0bb992fc56 update compensate CIC 2024-10-22 17:30:33 -04:00
Wassim KABALAN
31ca41b0a7 remove duplicate get_ode_fn 2024-10-22 13:38:16 -04:00
Wassim KABALAN
d28982eec7 Small fix 2024-10-22 13:00:25 -04:00
Wassim KABALAN
85cca44fb0 Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto 2024-10-22 12:59:50 -04:00
Wassim KABALAN
82b8f563a0 format 2024-10-22 12:57:34 -04:00
Wassim KABALAN
86233081e2 merge hugos LPT2 code 2024-10-22 12:57:05 -04:00
Wassim KABALAN
0433c615f3 quick fix in distributed 2024-10-22 12:15:37 -04:00
Wassim KABALAN
a8b194f326 quick fix in kernels 2024-10-22 12:15:17 -04:00
Wassim KABALAN
105568e8db painting.py no longer uses global mesh 2024-10-22 12:10:01 -04:00
Wassim KABALAN
80c56dced5 pm.py no longer uses global mesh 2024-10-22 12:09:50 -04:00
Wassim KABALAN
56ffd263f6 quick fix in distributed 2024-10-22 12:07:06 -04:00
Wassim KABALAN
a5b267bd63 kernels.py no longer uses global mesh 2024-10-22 11:05:21 -04:00
Wassim KABALAN
591ee32c55 global mesh no longer needed 2024-10-22 11:04:29 -04:00
Wassim KABALAN
38714cf65d add a strict dependency on jaxdecomp 2024-10-22 11:02:55 -04:00
Wassim KABALAN
9c94f994ff format 2024-10-21 14:00:14 -04:00
Wassim KABALAN
0ce7219ad8 make normal_field work with single controller 2024-10-21 13:59:55 -04:00
Wassim KABALAN
ff1c5e8362 remove fourrier_space in autoshmap 2024-10-21 13:57:26 -04:00
Wassim KABALAN
01b952701e update for latest jaxDecomp 2024-10-21 13:55:48 -04:00
Wassim KABALAN
afecb13cde revert single halo extent change 2024-10-20 09:58:24 -04:00
Wassim KABALAN
ab86699c88 merge with JZ 2024-10-18 14:59:40 -04:00
Wassim KABALAN
2ea05a1cd6 Update for jaxDecomp pure JAX 2024-10-18 20:45:08 +02:00
Wassim KABALAN
735699de77 roll back painting 2024-08-05 19:37:33 +02:00
Wassim KABALAN
8da3149581 fix formatting from main 2024-08-03 00:45:35 +02:00
Wassim KABALAN
783a97423e Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto 2024-08-03 00:23:54 +02:00
Wassim KABALAN
ece8c93540 format 2024-08-03 00:23:40 +02:00
Wassim KABALAN
ccbfee3615 add assert to make pyright happy 2024-08-02 21:21:59 +02:00
Wassim KABALAN
75604d2396 Shared operation in fourrier space now take inverted sharding axis for
slabs
2024-08-02 21:21:06 +02:00