jaxnet (pypi): Neural Nets for JAX
Keywords: data-science|deep-learning|jax|machine-learning|neural-networks|python