easyjax (pypi): Facilitates machine learning development for JAX.
Keywords: python|machine-learning|jax