elegy (pypi): Elegy is a Neural Networks framework based on Jax and Haiku.
Keywords: deep-learning|jax