maxtext.integration package

Subpackages

• maxtext.integration.tunix package
  • Subpackages
    • maxtext.integration.tunix.weight_mapping package
      • StandaloneVllmWeightMapping
      • Submodules
  • Submodules
    • maxtext.integration.tunix.tunix_adapter module
      • TunixMaxTextAdapter
    • maxtext.integration.tunix.utils module
      • VllmWeightMapping
• maxtext.integration.vllm package
  • Subpackages
    • maxtext.integration.vllm.maxtext_vllm_adapter package
      • register()
      • Submodules
    • maxtext.integration.vllm.torchax_converter package
      • Submodules
  • Submodules
    • maxtext.integration.vllm.maxtext_vllm_rollout module
    • maxtext.integration.vllm.setup module