趋近智
所有课程
3.1 并行处理原理介绍
3.2 JAX 中的设备管理
3.3 使用 pmap 进行单一程序多数据 (SPMD) 运算
3.4 使用 pmap 实现数据并行
3.5 集合通信原语 (psum, pmean 等)
3.6 pmap 中轴名称的处理
3.7 嵌套 pmap 与进阶分区
3.8 多主机编程简介
3.9 实践:分布式数据并行训练
© 2025 ApX Machine Learning