14 commits

Author	SHA1	Message	Date
EiffL	5108e56ee8	adds fix to make code jittablel	2022-05-17 23:37:55 +02:00
EiffL	5dc239927f	minor correction to gaussian smoothing	2022-05-17 23:02:01 +02:00
EiffL	e33504358d	adds utilities for simple lensing	2022-05-17 17:55:06 +02:00
EiffL	da2836f698	adding density plane cutting code	2022-05-17 11:26:38 +02:00
EiffL	03d3bc7927	adding function for doing 2d paintinng	2022-05-17 11:19:56 +02:00
EiffL	85e1c65951	adding hamiltonian gnn demo	2022-04-28 00:21:46 +02:00
EiffL	10b093f07f	fix minor issue	2022-03-26 03:03:59 +01:00
EiffL	3f9dfa504a	adding neural network	2022-03-26 02:59:39 +01:00
EiffL	3e1b3d8a3b	adding cic compensation tools	2022-03-26 00:06:34 +01:00
EiffL	079cebbdea	fix normalization of init cond	2022-03-25 22:34:13 +01:00
EiffL	607acf2e4f	Add utility to compute the power spectrum	2022-03-25 21:29:32 +01:00
EiffL	8543246f62	Adds demo and notebooks	2022-02-14 01:59:12 +01:00
EiffL	faec622152	adds growth functions from Chirag	2022-02-14 00:37:35 +01:00
EiffL	3507339663	Adds a trivial jaxpm implementation	2022-02-13 21:36:03 +01:00