16 commits

Author	SHA1	Message	Date
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	a8b194f326	quick fix in kernels	2024-10-22 12:15:17 -04:00
Wassim KABALAN	a5b267bd63	kernels.py no longer uses global mesh	2024-10-22 11:05:21 -04:00
Wassim KABALAN	01b952701e	update for latest jaxDecomp	2024-10-21 13:55:48 -04:00
Wassim KABALAN	ece8c93540	format	2024-08-03 00:23:40 +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	f25eb7d465	Adjust pencil type for frequencies	2024-07-28 13:44:03 +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
EiffL	319942a6bc	update formatting	2024-07-09 18:02:57 -04:00
EiffL	a742065ffd	adding example of distributed solution	2024-07-09 17:45:28 -04:00
EiffL	a2811c0606	Applying formatting	2024-07-09 14:54:34 -04:00
denise lanzieri	d3026f7849	PGD	2022-05-17 15:28:30 +02:00
EiffL	d8a5b702a9	adding cic compensation tools	2022-03-26 00:06:34 +01:00
EiffL	afdfcfa644	Adds a trivial jaxpm implementation	2022-02-13 21:36:03 +01:00