guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-04-24 11:50:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
84 commits 7 branches 5 tags 164 MiB
Commit graph

84 commits

This branch
This branch
All branches
Author SHA1 Message Date
Wassim KABALAN
be9a496161 Add draft example 2024-07-09 02:35:53 +02:00
Wassim KABALAN
1cade569af fixes for SPMD config 2024-07-09 02:35:37 +02:00
Wassim KABALAN
f3599e73da Fixes in solvers 2024-07-09 02:35:24 +02:00
Wassim KABALAN
a0be772f3c Add transpose for single GPU 2024-07-09 02:35:13 +02:00
Wassim KABALAN
d2fb1ee1e2 Adjust painting operators 2024-07-09 02:34:49 +02:00
Wassim KABALAN
6d8f130be7 Import at the start to register everything 2024-07-09 02:34:40 +02:00
Wassim KABALAN
e506dc2a7e add interpolate operator 2024-07-09 02:34:24 +02:00
Wassim KABALAN
c99b2c3ef4 remove fftk from kernels.py 2024-07-08 07:04:27 +02:00
Wassim KABALAN
da06f4dba8 Add solvers 2024-07-08 00:24:07 +02:00
Wassim KABALAN
8b9287184a Export operators 2024-07-08 00:23:53 +02:00
Wassim KABALAN
bc2612a198 Implement Ops 2024-07-08 00:23:35 +02:00
Wassim KABALAN
e708f5b176 Create operator infra 2024-07-08 00:23:12 +02:00
Wassim KABALAN
18d0bec2b5 add to gitignore 2024-07-08 00:22:44 +02:00
Wassim KABALAN
5b443e95c5 run precommit 2024-07-08 00:22:29 +02:00
Wassim KABALAN
38e875a7df add precommit 2024-07-08 00:21:33 +02:00
Wassim KABALAN
f49ed6fe70 Add design 2024-07-05 20:31:57 +02: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
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
Francois Lanusse
4a584b1d9c
Update jaxpm/painting.py 2022-05-18 10:22:21 +02:00
Francois Lanusse
9967dffd82
Merge pull request #12 from DifferentiableUniverseInitiative/PGD 
PGD
 2022-05-18 10:21:01 +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
eabac32e94 adding notebook demo 2022-05-17 17:56:41 +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
e8806e0520 merging upsteam 2022-04-28 00:22:57 +02:00
EiffL
d0a15e8c78 adding hamiltonian gnn demo 2022-04-28 00:21:46 +02:00
EiffL
1795319e4d adding hamiltonian gnn demo 2022-04-28 00:20:44 +02:00
EiffL
687aad6571 fix power spec 2022-03-27 04:39:07 +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