inox (pypi): Stainless neural networks in JAX
Keywords: jax|neural|networks|deep learning