/jax-cosmos

Library of usefull helper functions for dealing with voxels and particles in JAX.

jax-cosmos