World Engine
  • Quickstart
    • Installation
    • Basic Usage
      • Understanding CtrlInput
      • Using WorldEngine
        • Loading a Model
        • Setting the Prompt
        • Appending Frames
        • Generating Frames
        • Resetting State
      • Complete Example: Interactive Pygame Application
    • Next Steps
  • Main API Reference
    • WorldEngine
      • WorldEngine
        • WorldEngine.__init__()
        • WorldEngine.reset()
        • WorldEngine.set_prompt()
        • WorldEngine.append_frame()
        • WorldEngine.gen_frame()
        • WorldEngine.prep_inputs()
    • CtrlInput
      • CtrlInput
        • CtrlInput.button
        • CtrlInput.mouse
        • CtrlInput.scroll_wheel
        • CtrlInput.__init__()
  • Quantization
    • world_engine.quantize
      • fp4_linear()
      • FP4Linear
        • FP4Linear.__init__()
        • FP4Linear.forward()
      • FP8W8A8Linear
        • FP8W8A8Linear.__init__()
        • FP8W8A8Linear.forward()
      • FP8Linear
        • FP8Linear.__init__()
        • FP8Linear.forward()
      • quantize_model()
  • Modeling Specifics
    • Internal Model Components
      • world_engine.model.base_model
        • BaseModel
      • world_engine.model.world_model
        • PromptEncoder
        • rms_norm()
        • CFG
        • MLP
        • ControllerInputEmbedding
        • MLPFusion
        • CondHead
        • WorldDiTBlock
        • WorldDiT
        • WorldModel
      • world_engine.model.nn
        • NoCastModule
        • rms_norm()
        • MLP
        • AdaLN
        • ada_rmsnorm()
        • ada_gate()
        • NoiseConditioner
      • world_engine.model.attn
        • RoPE
        • OrthoRoPE
        • Attn
        • CrossAttention
      • world_engine.model.kv_cache
        • make_block_mask()
        • LayerKVCache
        • StaticKVCache
    • Patch Model
      • world_engine.patch_model
        • CachedDenoiseStepEmb
        • CachedCondHead
        • patch_cached_noise_conditioning()
        • MergedQKVAttn
        • patch_Attn_merge_qkv()
        • SplitMLPFusion
        • patch_MLPFusion_split()
        • apply_inference_patches()
    • Autoencoder Neural Network Components
      • world_engine.ae_nn
        • WeightNormConv2d()
        • ResBlock
        • LandscapeToSquare
        • Downsample
        • DownBlock
        • SpaceToChannel
        • ChannelAverage
        • SquareToLandscape
        • Upsample
        • UpBlock
        • ChannelToSpace
        • ChannelDuplication
        • Encoder
        • Decoder
        • AutoEncoder
World Engine
  • Overview: module code

All modules for which code is available

  • world_engine.ae_nn
  • world_engine.model.attn
  • world_engine.model.base_model
  • world_engine.model.kv_cache
  • world_engine.model.nn
  • world_engine.model.world_model
  • world_engine.patch_model
  • world_engine.quantize
  • world_engine.world_engine

© Copyright 2026, World Engine Team.

Built with Sphinx using a theme provided by Read the Docs.