Keras

Keras is a multi-framework deep learning API (written in Python) that runs on top of JAX, TensorFlow, or PyTorch.

Keras 3 is the multi-framework deep learning API, designed for human-centric development: it focuses on reducing cognitive load and enabling fast experimentation. It provides a simple, consistent interface for building models, from basic Sequential stacks to complex Functional API architectures. The framework offers industry-strength performance and scalability, leveraging backends like JAX for optimized training on GPUs and TPUs. Organizations like NASA, YouTube, and Waymo utilize Keras for production-grade applications, proving its power and reliability across computer vision and NLP domains.

https://keras.io