maxtext.integration.tunix package

Subpackages

• maxtext.integration.tunix.weight_mapping package
  • StandaloneVllmWeightMapping
  • Submodules
    • maxtext.integration.tunix.weight_mapping.deepseek3 module
      • DEEPSEEK_VLLM_MAPPING
    • maxtext.integration.tunix.weight_mapping.gpt_oss module
      • GPT_OSS_VLLM_MAPPING
    • maxtext.integration.tunix.weight_mapping.llama3 module
      • LLAMA3_VLLM_MAPPING
    • maxtext.integration.tunix.weight_mapping.qwen2 module
      • QWEN2_VLLM_MAPPING
    • maxtext.integration.tunix.weight_mapping.qwen3 module
      • QWEN3_VLLM_MAPPING

Submodules

• maxtext.integration.tunix.tunix_adapter module
  • TunixMaxTextAdapter
    • TunixMaxTextAdapter.to_hf_mappings()
    • TunixMaxTextAdapter.to_hf_transpose_keys()
    • TunixMaxTextAdapter.to_hf_hook_fns()
    • TunixMaxTextAdapter.lora_to_hf_mappings()
• maxtext.integration.tunix.utils module
  • VllmWeightMapping
    • VllmWeightMapping.to_hf_mapping()
    • VllmWeightMapping.to_hf_transpose_keys()
    • VllmWeightMapping.to_hf_hook_fns()
    • VllmWeightMapping.lora_to_hf_mappings()
    • VllmWeightMapping.convert_hf_map_to_sharding_map()