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

update code

Browse source
This commit is contained in:
Wassim Kabalan 2024-12-06 18:56:24 +01:00
parent e0c118a540
commit 21373b89ee
7 changed files with 84 additions and 100 deletions
Download patch file Download diff file Expand all files Collapse all files

1
jaxpm/kernels.py
View file

@ -1,5 +1,4 @@
import jax.numpy as jnp
import jax_cosmo as jc
import numpy as np
from jax.lib.xla_client import FftType
from jax.sharding import PartitionSpec as P