guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-04-18 17:10:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
189 commits 7 branches 5 tags 164 MiB
Commit graph

189 commits

This branch
This branch
All branches
Author SHA1 Message Date
Wassim Kabalan
47c69c604e add numpy explicitly as dependency for tests 2024-12-08 23:25:32 +01:00
Wassim Kabalan
b264da5b17 JAX version is 0.4.35 until Diffrax new release 2024-12-08 23:13:31 +01:00
Wassim Kabalan
d8c68ace7a Add aboucaud comments 2024-12-08 23:11:11 +01:00
Wassim Kabalan
adaf7d236d fix deprecated FftType in jaxpm.kernels 2024-12-08 22:54:52 +01:00
Wassim Kabalan
5d34d3c3a8 add test workflow 2024-12-08 22:46:57 +01:00
Wassim Kabalan
ac4ef9e1fa update test dependencies 2024-12-08 22:46:46 +01:00
Wassim Kabalan
97f39bd051 format tests 2024-12-08 22:46:28 +01:00
Wassim Kabalan
af29c4005d Format 2024-12-08 22:45:09 +01:00
Wassim Kabalan
7823fdaf98 Upgrade setup.py to pyproject 2024-12-06 18:57:16 +01:00
Wassim Kabalan
f70583b5fd add tests 2024-12-06 18:56:57 +01:00
Wassim Kabalan
36ef18e3d0 update notebooks 2024-12-06 18:56:35 +01:00
Wassim Kabalan
21373b89ee update code 2024-12-06 18:56:24 +01:00
Wassim Kabalan
e0c118a540 By default use absoulute painting with 2024-12-05 18:23:18 +01:00
Wassim Kabalan
c1b276d224 Put plotting utils in package 2024-12-05 18:22:34 +01:00
Wassim Kabalan
b32014b7ea Add finite difference laplace kernel + powerspec functions from Hugo 
Co-authored-by: Hugo Simonfroy <hugo.simonfroy@gmail.com>
 2024-12-05 18:21:24 +01:00
Wassim KABALAN
435c7c848f Put plotting in package 2024-11-14 13:52:59 +01:00
Wassim KABALAN
0946842fe5 Adding animation notebook 2024-11-10 21:00:08 +01:00
Wassim KABALAN
12c74e2601 ability to choose snapshots number with MultiHost script 2024-11-10 20:59:54 +01:00
Wassim KABALAN
ad4566620f clarifying the halo size dependence on the box size 2024-11-10 20:58:29 +01:00
Wassim KABALAN
2472a5d9df clarifying why cic_paint_dx is slower 2024-11-10 20:58:03 +01:00
Wassim KABALAN
6256fbadb1 forgot particles in multi host 2024-10-30 02:30:41 +01:00
Wassim KABALAN
42d8e89de4 fix small error 2024-10-30 02:03:54 +01:00
Wassim KABALAN
a067954d0a update readme 2024-10-30 02:01:05 +01:00
Wassim KABALAN
f0c43f8d76 add readme for tutorials 2024-10-30 02:01:00 +01:00
Wassim KABALAN
a030ec47c6 delete outdated design file 2024-10-30 01:59:07 +01:00
Wassim KABALAN
72457d6c37 update example notebooks 2024-10-30 01:58:47 +01:00
Wassim KABALAN
4da4c66472 remove script folder 2024-10-30 01:58:17 +01:00
Wassim KABALAN
e9529d35f8 Weights can be traced 2024-10-30 01:57:56 +01:00
Wassim KABALAN
b09580d59e Allow applying weights with relative cic paint 2024-10-30 01:57:32 +01:00
Wassim KABALAN
b3a264ad53 use float64 for enmeshing 2024-10-30 01:56:29 +01:00
Wassim KABALAN
2ad035a75b use jax interp instead of jax_cosmo 2024-10-30 01:56:04 +01:00
Wassim KABALAN
f3b431aa74 add a user facing function to create uniform particle grid 2024-10-30 01:55:47 +01:00
Wassim KABALAN
a757b62f4b get local shape and zeros can be used by users 2024-10-30 01:55:20 +01:00
Wassim KABALAN
19011d0712 apply formating 2024-10-27 03:50:34 +01:00
Wassim KABALAN
c93894f561 update notebooks 2024-10-27 03:49:07 +01:00
Wassim KABALAN
b4fdb74660 jit cic_paint 2024-10-27 03:48:50 +01:00
Wassim KABALAN
d62c38f457 fix code in LPT2 2024-10-27 03:48:38 +01:00
Wassim KABALAN
cc4f310508 add distributed zeros 2024-10-27 03:48:17 +01:00
Wassim KABALAN
4342279817 apply formatting 2024-10-27 00:52:14 +02:00
Wassim KABALAN
11f7e90066 Merge remote-tracking branch 'upstream/ASKabalan/jaxdecomp_proto' into ASKabalan/jaxdecomp_proto 2024-10-27 00:51:16 +02:00
Wassim KABALAN
49dd18a3f8 add notebook examples 2024-10-26 22:49:17 +02:00
Wassim KABALAN
0c96a4dc10 delete old slurms 2024-10-26 22:48:38 +02:00
Wassim KABALAN
ff8856d2bc use jnp interp instead of jc interp 2024-10-26 22:48:38 +02:00
Wassim KABALAN
d2f1eb2fa4 fix painting issue with read_cic 2024-10-26 22:48:38 +02:00
Francois Lanusse
0f833f0cb4 remove deprecated stuff 2024-10-26 22:48:38 +02:00
Wassim KABALAN
69c35d1aa8 Merge remote-tracking branch 'upstream/main' into ASKabalan/jaxdecomp_proto 2024-10-25 11:04:55 +02:00
Wassim KABALAN
8e8e8964be jaxDecomp pfft3d promotes to complex automatically 2024-10-25 10:15:43 +02:00
Wassim KABALAN
5d4f438e92 update PGDCorrection and neural ode to use new fft3d 2024-10-25 10:14:29 +02:00
Wassim KABALAN
505f2ec286 remove experimental distributed code 2024-10-25 10:05:33 +02:00
Francois Lanusse
a848d43b56
remove deprecated stuff 2024-10-24 16:36:41 -04:00