105 commits

This branch

This branch

All branches

Author	SHA1	Message	Date
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
Wassim KABALAN	5775a37550	implement distributed optimized cic_paint	2024-07-18 12:39:15 +02:00
Wassim KABALAN	e62cd84cbd	apply formatting	2024-07-18 12:38:42 +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
Francois Lanusse	2db7c43ab8	Merge pull request #16 from DifferentiableUniverseInitiative/u/EiffL/distributed ... U/eiff l/distributed	2022-10-22 07:23:51 -04: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
EiffL	2da21d8b5e	Added prototype distributed LPT	2022-10-19 17:37:38 -07:00
EiffL	61578217fe	Added example code to compute distributed fft	2022-10-17 22:07:12 -07: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	5c4f75a102	fixes the lensing demo	2022-05-18 15:43:45 +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