maxtext.integration.vllm package

Subpackages

• maxtext.integration.vllm.maxtext_vllm_adapter package
  • register()
  • Submodules
    • maxtext.integration.vllm.maxtext_vllm_adapter.adapter module
      • next_power_of_two()
      • generate_maxtext_config()
      • MaxTextForCausalLM
• maxtext.integration.vllm.torchax_converter package
  • Submodules
    • maxtext.integration.vllm.torchax_converter.base module
      • timer()
      • BaseMaxTextToVLLMConverter
    • maxtext.integration.vllm.torchax_converter.gemma4_moe module
      • Gemma4MaxTextToVLLMConverter
    • maxtext.integration.vllm.torchax_converter.qwen3_moe module
      • Qwen3MaxTextToVLLMConverter
    • maxtext.integration.vllm.torchax_converter.validate_converter module

Submodules

• maxtext.integration.vllm.maxtext_vllm_rollout module
• maxtext.integration.vllm.setup module