Skip to main content
Ctrl+K

Slapo Documentation

Getting Started

  • Installation
  • Quick Start

Tutorials

  • Optimize Attention Module on A Single Device
  • Optimize MLP Module on Multi-Device
  • Debugging with Print

Reference

  • Python API
    • slapo
    • slapo.schedule
    • slapo.initialization
    • slapo.pattern
    • slapo.pipeline
    • slapo.tracer
    • slapo.random
    • slapo.framework_dialect
      • slapo.framework_dialect.registry
      • slapo.framework_dialect.deepspeed
        • slapo.framework_dialect.deepspeed.engine
        • slapo.framework_dialect.deepspeed.pipeline
    • slapo.model_schedule
      • slapo.model_schedule.api
    • slapo.op
      • slapo.op.attention
      • slapo.op.cross_entropy
      • slapo.op.fused_bias
      • slapo.op.linear
      • slapo.op.mlp
  • Index
  • .rst

slapo.framework_dialect

slapo.framework_dialect¶

Submodules

  • slapo.framework_dialect.registry
  • slapo.framework_dialect.deepspeed
    • slapo.framework_dialect.deepspeed.engine
    • slapo.framework_dialect.deepspeed.pipeline

previous

slapo.random

next

slapo.framework_dialect.registry

By Amazon

© Copyright 2023, Amazon.