ninjax (pypi): General Modules for JAX
Keywords: deep-learning|jax