About Keras Getting started Developer guides Code examples Keras 3 API documentation 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 Utilities Keras 2 API documentation KerasTuner: Hyperparam Tuning KerasHub: Pretrained Models
► Keras 3 API documentation / Multi-device distribution

Multi-device distribution

LayoutMap API

  • LayoutMap class
  • DeviceMesh class
  • TensorLayout class
  • distribute_tensor function

DataParallel API

  • DataParallel class

ModelParallel API

  • ModelParallel class

ModelParallel API

  • ModelParallel class

Distribution utilities

  • set_distribution function
  • distribution function
  • list_devices function
  • initialize function
Multi-device distribution
LayoutMap API
DataParallel API
ModelParallel API
ModelParallel API
Distribution utilities
Terms | Privacy