22 commits

Author	SHA1	Message	Date
Wassim Kabalan	21373b89ee	update code	2024-12-06 18:56:24 +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	ff8856d2bc	use jnp interp instead of jc interp	2024-10-26 22:48:38 +02:00
Wassim KABALAN	0bb992fc56	update compensate CIC	2024-10-22 17:30:33 -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	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
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	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