maxtext.integration.vllm.torchax_converter.base module
Shared base classes for MaxText to vLLM converters.
-
maxtext.integration.vllm.torchax_converter.base.timer(name)[source]
-
class maxtext.integration.vllm.torchax_converter.base.BaseMaxTextToVLLMConverter(config, mesh)[source]
Bases: ABC
Shared converter contract for MaxText to vLLM weight conversion.
-
convert(model_state)[source]
Convert a MaxText model state into vLLM weight tensors.
- Parameters:
model_state (dict)
