mlax-nn (pypi): A pure functional machine learning library build on top of Google JAX.
Keywords: deep-learning|jax|machine-learning|neural-networks