71 commits

Author	SHA1	Message	Date
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	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
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
Wassim KABALAN	4f508b7cb6	format	2024-07-18 13:16:14 +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
Wassim KABALAN	5775a37550	implement distributed optimized cic_paint	2024-07-18 12:39:15 +02:00
Wassim KABALAN	ac86468c7c	add halo exchange and slice pad	2024-07-10 00:25:52 +02:00
EiffL	319942a6bc	update formatting	2024-07-09 18:02:57 -04:00
EiffL	6408aff1de	put back old functgion	2024-07-09 18:00:35 -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
Francois Lanusse	5f463450d1	Adding option to have weights in the 3d cic paint	2023-05-05 19:00:08 +02:00
EiffL	d9578b1220	removed old files	2022-10-22 07:17:44 -04:00
EiffL	a6096ae04b	moved xmap stuff to experimental	2022-10-22 07:17:29 -04:00
EiffL	a2e8017bc8	added things	2022-10-20 21:22:39 -07:00
EiffL	953fc48cc0	update data	2022-10-20 21:03:15 -07:00
EiffL	7f3de80f7a	adding distributed ops	2022-10-20 23:05:39 -04:00
denise lanzieri	b26cfcd01a	creoss correlation function	2022-06-18 18:23:46 +02:00
denise lanzieri	f185b8b913	few adjustments to PGD correction	2022-06-13 17:17:19 +02:00
denise lanzieri	5b72a73448	neural ode added	2022-06-11 14:28:30 +02:00
Francois Lanusse	101ef80d9c	Merge pull request #13 from DifferentiableUniverseInitiative/u/EiffL/lensing ... U/eiff l/lensing	2022-05-18 15:55:40 +02:00
EiffL	1b7c797286	changes definition of lensplanes	2022-05-18 15:40:11 +02:00
Francois Lanusse	e7fa6c9e57	Merge pull request #11 from DifferentiableUniverseInitiative/u/EiffL/lensing ... Adds basic utilities for Born lensing	2022-05-18 14:35:06 +02:00
Francois Lanusse	4a584b1d9c	Update jaxpm/painting.py	2022-05-18 10:22:21 +02:00