134 commits

This branch

This branch

All branches

Author	SHA1	Message	Date
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	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	ebe89e2ec6	Merge pull request #23 from hsimonfroy/main ... 2lpt	2024-08-05 19:38:35 +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	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
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
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