Overview

The Keras ecosystem offers many on-ramps to JAX. Use our classic and well-loved API patterns such as `model.fit(...)` to author and train models using JAX. Want to start with a pre-trained model? Try KerasHub, our catalog of 100+ model architectures with support for thousands of HuggingFace checkpoints. Fine-tune and optimize them using LoRA and qLoRA with JAX (or your choice of backend). We’ll also introduce Keras Recommenders, the latest addition to the Keras ecosystem.