Commit graph

79 commits

Author SHA1 Message Date
Hugo Simonfroy
9b21eed3b5 2lpt, get_ode, invlaplace, docstrings 2024-07-31 00:46:53 +02:00
Francois Lanusse
b0b793e766 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]
0d3bdda3ab docs: update .all-contributorsrc [skip ci] 2024-07-19 14:51:26 +00:00
allcontributors[bot]
ad2ce8d825 docs: update README.md [skip ci] 2024-07-19 14:51:25 +00:00
Francois Lanusse
0f81a89d57 Merge pull request #14 from DifferentiableUniverseInitiative/neural_ode
Fourier-space Neural Network  scheme
2024-07-19 10:50:36 -04:00
Francois Lanusse
b949827e92 Update jaxpm/pm.py 2024-07-19 10:49:51 -04:00
Francois Lanusse
9a279d2d6c Merge branch 'main' into neural_ode 2024-07-19 10:48:09 -04:00
EiffL
f28442bb48 Applying formatting 2024-07-09 14:54:34 -04:00
Francois Lanusse
835fa89aec Adding option to have weights in the 3d cic paint 2023-05-05 19:00:08 +02:00
Francois Lanusse
d4673d2955 Merge pull request #16 from DifferentiableUniverseInitiative/u/EiffL/distributed
U/eiff l/distributed
2022-10-22 07:23:51 -04:00
EiffL
ada6ff9b70 removed old files 2022-10-22 07:17:44 -04:00
EiffL
aa90804536 moved xmap stuff to experimental 2022-10-22 07:17:29 -04:00
EiffL
70600fb7f7 added things 2022-10-20 21:22:39 -07:00
EiffL
ebab27b4c3 update data 2022-10-20 21:03:15 -07:00
EiffL
137f4e5099 adding distributed ops 2022-10-20 23:05:39 -04:00
EiffL
1cb2739366 Added prototype distributed LPT 2022-10-19 17:37:38 -07:00
EiffL
64894726e7 Added example code to compute distributed fft 2022-10-17 22:07:12 -07:00
denise lanzieri
84b79af7f8 creoss correlation function 2022-06-18 18:23:46 +02:00
denise lanzieri
8b885450a8 few adjustments to PGD correction 2022-06-13 17:17:19 +02:00
denise lanzieri
d8a1dbe210 neural ode added 2022-06-11 14:28:30 +02:00
Francois Lanusse
20fc4a5562 Merge pull request #13 from DifferentiableUniverseInitiative/u/EiffL/lensing
U/eiff l/lensing
2022-05-18 15:55:40 +02:00
EiffL
cd7a48e463 fixes the lensing demo 2022-05-18 15:43:45 +02:00
EiffL
301bd37647 changes definition of lensplanes 2022-05-18 15:40:11 +02:00
Francois Lanusse
0991789553 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
ff5fe8069e Update jaxpm/painting.py 2022-05-18 10:22:21 +02:00
Francois Lanusse
e93aa07f93 Merge pull request #12 from DifferentiableUniverseInitiative/PGD
PGD
2022-05-18 10:21:01 +02:00
Denise Lanzieri
3df0c05d29 Update jaxpm/pm.py
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
2022-05-18 09:59:59 +02:00
Denise Lanzieri
dd71d359e0 Update jaxpm/pm.py
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
2022-05-18 09:58:46 +02:00
Denise Lanzieri
77827fcf44 Update jaxpm/pm.py
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
2022-05-18 09:58:34 +02:00
Denise Lanzieri
85b2f4f097 Update jaxpm/pm.py
Co-authored-by: Francois Lanusse <EiffL@users.noreply.github.com>
2022-05-18 09:55:31 +02:00
EiffL
60309e1490 small fix 2022-05-17 23:49:12 +02:00
EiffL
aff8db56f5 change impor 2022-05-17 23:42:57 +02:00
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
ca5f26c93f adding notebook demo 2022-05-17 17:56:41 +02:00
EiffL
e33504358d adds utilities for simple lensing 2022-05-17 17:55:06 +02:00
denise lanzieri
6b6b414195 PGD 2022-05-17 15:28:30 +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
15e2c5cc6e merging upsteam 2022-04-28 00:22:57 +02:00
EiffL
85e1c65951 adding hamiltonian gnn demo 2022-04-28 00:21:46 +02:00
EiffL
fdb25fb661 adding hamiltonian gnn demo 2022-04-28 00:20:44 +02:00
EiffL
3e1aa2d4e2 fix power spec 2022-03-27 04:39:07 +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
211a0faf94 Adding demo CAMELS notebook 2022-03-21 09:39:02 +01:00
Francois Lanusse
cdea4c37ee Update README.md 2022-02-14 11:56:27 +01:00