Elegy is a Neural Networks framework based on Jax and inspired by Keras. - View it on GitHub