31 commits

Author	SHA1	Message	Date
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
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
Denise Lanzieri	5394abbed1	Update jaxpm/pm.py ... Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>	2022-05-18 09:59:59 +02:00
Denise Lanzieri	782c9dbc6c	Update jaxpm/pm.py ... Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>	2022-05-18 09:58:46 +02:00
Denise Lanzieri	8266f11369	Update jaxpm/pm.py ... Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>	2022-05-18 09:58:34 +02:00
Denise Lanzieri	b737631346	Update jaxpm/pm.py ... Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>	2022-05-18 09:55:31 +02:00
EiffL	0f6fb39966	small fix	2022-05-17 23:49:12 +02:00
EiffL	a52287cd97	change impor	2022-05-17 23:42:57 +02:00
EiffL	88cff99736	adds fix to make code jittablel	2022-05-17 23:37:55 +02:00
EiffL	80eb8cb2a3	minor correction to gaussian smoothing	2022-05-17 23:02:01 +02:00
EiffL	0008f8549b	adds utilities for simple lensing	2022-05-17 17:55:06 +02:00
denise lanzieri	d3026f7849	PGD	2022-05-17 15:28:30 +02:00
EiffL	e45bab2ecf	adding density plane cutting code	2022-05-17 11:26:38 +02:00
EiffL	e188d5efb6	adding function for doing 2d paintinng	2022-05-17 11:19:56 +02:00
EiffL	d0a15e8c78	adding hamiltonian gnn demo	2022-04-28 00:21:46 +02:00
EiffL	5195a28582	fix minor issue	2022-03-26 03:03:59 +01:00
EiffL	907dc42bdf	adding neural network	2022-03-26 02:59:39 +01:00
EiffL	d8a5b702a9	adding cic compensation tools	2022-03-26 00:06:34 +01:00
EiffL	5549568295	fix normalization of init cond	2022-03-25 22:34:13 +01:00
EiffL	b08641d51d	Add utility to compute the power spectrum	2022-03-25 21:29:32 +01:00
EiffL	25bde5581e	Adds demo and notebooks	2022-02-14 01:59:12 +01:00
EiffL	c9a42686eb	adds growth functions from Chirag	2022-02-14 00:37:35 +01:00
EiffL	afdfcfa644	Adds a trivial jaxpm implementation	2022-02-13 21:36:03 +01:00