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

Python API

Python API¶

Python API

  • slapo
  • slapo.schedule
  • slapo.initialization
  • slapo.pattern
  • slapo.pipeline
  • slapo.tracer
  • slapo.random
  • slapo.framework_dialect
  • slapo.model_schedule
  • slapo.op

previous

Debugging with Print

next

slapo

By Amazon

© Copyright 2023, Amazon.