Wassim KABALAN
|
86233081e2
|
merge hugos LPT2 code
|
2024-10-22 12:57:05 -04:00 |
|
Wassim KABALAN
|
2f509932f5
|
update demo
|
2024-10-22 12:16:28 -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
|
4d944f01f2
|
update demo script
|
2024-10-22 12:10:18 -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
|
a160a3faa9
|
update to leapfrog
|
2024-10-22 09:03:41 -04:00 |
|
Wassim KABALAN
|
5a587fd402
|
merge bench_pm
|
2024-10-21 14:04:09 -04:00 |
|
Wassim KABALAN
|
375f2048e4
|
make distributed pm work in single controller
|
2024-10-21 14:03:48 -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
|
8da3149581
|
fix formatting from main
|
2024-08-03 00:45:35 +02:00 |
|
Wassim KABALAN
|
02754cf452
|
added github workflow
|
2024-08-03 00:44:05 +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
|
831291c1f9
|
bench
|
2024-08-02 23:39:09 +02:00 |
|
Wassim KABALAN
|
9af4659c81
|
add PMWD test
|
2024-08-02 21:22:56 +02:00 |
|
Wassim KABALAN
|
aebc3e72c0
|
adjust test for hpc-plotter
|
2024-08-02 21:22:45 +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 |
|
Wassim KABALAN
|
f25eb7d465
|
Adjust pencil type for frequencies
|
2024-07-28 13:44:03 +02:00 |
|
Wassim KABALAN
|
1f2035176f
|
Update scripts/distributed_pm.py
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
|
2024-07-19 19:25:26 +02:00 |
|
Francois Lanusse
|
2bb0f9e9a9
|
Merge pull request #22 from DifferentiableUniverseInitiative/all-contributors/add-dlanzieri
docs: add dlanzieri as a contributor for code
|
2024-07-19 10:52:37 -04:00 |
|
allcontributors[bot]
|
b4adc119b6
|
docs: update .all-contributorsrc [skip ci]
|
2024-07-19 14:51:26 +00:00 |
|
allcontributors[bot]
|
38270f3358
|
docs: update README.md [skip ci]
|
2024-07-19 14:51:25 +00:00 |
|
Francois Lanusse
|
d2ab0fe6e9
|
Merge pull request #14 from DifferentiableUniverseInitiative/neural_ode
Fourier-space Neural Network scheme
|
2024-07-19 10:50:36 -04: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
|
5b7f595c32
|
Visualize only final field
|
2024-07-18 17:05:22 +02:00 |
|
Wassim KABALAN
|
0216837033
|
update benchmark and add slurm
|
2024-07-18 17:04:52 +02:00 |
|
Wassim KABALAN
|
ed8cf8e532
|
add lpt2
|
2024-07-18 17:04:05 +02:00 |
|
Wassim KABALAN
|
1f6b9c3217
|
times in ms
|
2024-07-18 13:23:53 +02:00 |
|
Wassim KABALAN
|
5f6d42eaeb
|
add optimised bench script
|
2024-07-18 13:16:37 +02:00 |
|
Wassim KABALAN
|
4f508b7cb6
|
format
|
2024-07-18 13:16:14 +02:00 |
|
Wassim KABALAN
|
82be56836a
|
add example
|
2024-07-18 12:45:56 +02:00 |
|
Wassim KABALAN
|
abde5439f6
|
Return normal order frequencies for single GPU
|
2024-07-18 12:44:18 +02:00 |
|
Wassim KABALAN
|
c81d4d2336
|
Wrap interpolation function to avoid all gather
|
2024-07-18 12:43:52 +02:00 |
|
Wassim KABALAN
|
7f48cfa8af
|
Fix seed for distributed normal
|
2024-07-18 12:43:24 +02:00 |
|
Wassim KABALAN
|
7501b5bc6d
|
Use new cic_paint with halo
|
2024-07-18 12:42:07 +02:00 |
|