Multi-device distribution

• Original Link : https://keras.io/api/distribution/
• Last Checked at : 2024-11-25

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