A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch. - View it on GitHub