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 |
|
Wassim KABALAN
|
8c5bd76c33
|
fix painting issue with slabs
|
2024-08-02 21:20:15 +02:00 |
|
Hugo Simonfroy
|
d85867e88d
|
2lpt, get_ode, invlaplace, docstrings
|
2024-07-31 00:46:53 +02:00 |
|
Wassim KABALAN
|
f25eb7d465
|
Adjust pencil type for frequencies
|
2024-07-28 13:44:03 +02:00 |
|
Francois Lanusse
|
9cbdf18932
|
Update jaxpm/pm.py
|
2024-07-19 10:49:51 -04:00 |
|
Francois Lanusse
|
945dae4f80
|
Merge branch 'main' into neural_ode
|
2024-07-19 10:48:09 -04:00 |
|
Wassim KABALAN
|
ed8cf8e532
|
add lpt2
|
2024-07-18 17:04:05 +02:00 |
|