guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-29 08:31:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

format

Browse source
This commit is contained in:
Wassim KABALAN 2024-07-18 13:15:39 +02:00
parent 82be56836a
commit 4f508b7cb6
3 changed files with 54 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

1
jaxpm/pm.py
View file

@ -38,6 +38,7 @@ def pm_forces(positions, mesh_shape=None, delta=None, r_split=0, halo_size=0):
return forces
def lpt(cosmo, initial_conditions, a, halo_size=0):
"""
Computes first order LPT displacement