keras.io
网页
综合
Multi-GPU distributed training with JAX
…sharding APIs to train Keras models, with minimal changes to your code, on multiple GPUs or TPUS (typically 2 to 16) installed on a single machine (single host, multi-device training). This …
2026-11-15
自动蜘蛛
相关度 2.3