Models API Layers API Callbacks API Ops API Optimizers Metrics Losses Data loading Built-in small datasets Keras Applications Mixed precision Multi-device distribution LayoutMap API DataParallel API ModelParallel API ModelParallel API Distribution utilities RNG API Rematerialization Utilities