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.op

slapo.op¶

Submodules

  • slapo.op.attention
  • slapo.op.cross_entropy
  • slapo.op.fused_bias
  • slapo.op.linear
  • slapo.op.mlp

previous

slapo.model_schedule.api

next

slapo.op.attention

By Amazon

© Copyright 2023, Amazon.