DeepSpeed
  • 训练设置
  • 推理设置
  • Training API
  • Inference API
  • 模型检查点
  • 激活检查点
  • ZeRO
  • 专家混合模型 (MoE)
  • Transformer Kernels
  • 管道并行
  • 优化器
  • 学习率调度器
  • Flops Profiler
  • 自动调优
  • 内存需求
  • 监控
DeepSpeed
  • Overview: module code

所有可用代码的模块

  • deepspeed.checkpoint.utils
  • deepspeed.inference.config
  • deepspeed.moe.layer
  • deepspeed.monitor.config
  • deepspeed.ops.adam.cpu_adam
  • deepspeed.ops.adam.fused_adam
  • deepspeed.ops.lamb.fused_lamb
  • deepspeed.runtime.lr_schedules
  • deepspeed.runtime.pipe.module
  • deepspeed.runtime.pipe.schedule
  • deepspeed.runtime.pipe.topology
  • deepspeed.runtime.zero.config
  • deepspeed.runtime.zero.offload_config
  • deepspeed.runtime.zero.stage3
  • deepspeed.runtime.zero.stage_1_and_2
  • deepspeed.utils.tensor_fragment
  • deepspeed.utils.zero_to_fp32

© 版权所有 2020, Microsoft.

Built with Sphinx using a theme provided by Read the Docs.