文档—沐曦开发者

目录

• 1. 概述
• 2. 快速安装
  • 2.1. 基于pip安装
    • 2.1.1. 环境准备
    • 2.1.2. 开始安装
    • 2.1.3. 验证安装
    • 2.1.4. 如何卸载
  • 2.2. 使用Docker运行
• 3. 功能支持
  • 3.1. apex.fused_dense.FusedDense
  • 3.2. apex.transformer.functional.FusedScaleMaskSoftmax
  • 3.3. apex.transformer.functional.fused_apply_rotary_pos_emb
  • 3.4. apex.transformer.functional.fused_apply_rotary_pos_emb_cached
  • 3.5. apex.transformer.functional.fused_apply_rotary_pos_emb_thd
  • 3.6. apex.normalization.FusedLayerNorm
  • 3.7. apex.normalization.FusedRMSNorm
  • 3.8. apex.normalization.MixedFusedLayerNorm
  • 3.9. apex.normalization.MixedFusedRMSNorm
  • 3.10. apex.contrib.layer_norm.FastLayerNorm