solstice-jax (pypi): Solstice, a library for creating and scaling experiments in JAX.
Keywords: deep-learning|jax