Framework-specific API

• pyTorch
  • Modules
    • Linear
      • Linear.forward()
    • LayerNorm
    • LayerNormLinear
      • LayerNormLinear.forward()
    • LayerNormMLP
      • LayerNormMLP.forward()
    • TransformerLayer
      • TransformerLayer.forward()
  • Functions
    • fp8_autocast()
    • checkpoint()