guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-04-27 20:52:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use jax interp instead of jax_cosmo

Browse source
This commit is contained in:
Wassim KABALAN 2024-10-30 01:56:04 +01:00
parent f3b431aa74
commit 2ad035a75b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
jaxpm/growth.py
View file

@ -1,6 +1,6 @@
import jax.numpy as np
from jax.numpy import interp
from jax_cosmo.background import *
from jax_cosmo.scipy.interpolate import interp
from jax_cosmo.scipy.ode import odeint