Supported Models

The following is a table of supported models for the PyTorch backend:

Architecture	Model	HuggingFace Example
BertForSequenceClassification	BERT-based	textattack/bert-base-uncased-yelp-polarity
DeciLMForCausalLM	Nemotron	nvidia/Llama-3_1-Nemotron-51B-Instruct
DeepseekV3ForCausalLM	DeepSeek-V3	deepseek-ai/DeepSeek-V3
Exaone4ForCausalLM	EXAONE 4.0	LGAI-EXAONE/EXAONE-4.0-32B
Gemma3ForCausalLM	Gemma 3	google/gemma-3-1b-it
LlamaForCausalLM	Llama 3.1, Llama 3, Llama 2, LLaMA	meta-llama/Meta-Llama-3.1-70B
Llama4ForConditionalGeneration	Llama 4	meta-llama/Llama-4-Scout-17B-16E-Instruct
MistralForCausalLM	Mistral	mistralai/Mistral-7B-v0.1
MixtralForCausalLM	Mixtral	mistralai/Mixtral-8x7B-v0.1
MllamaForConditionalGeneration	Llama 3.2	meta-llama/Llama-3.2-11B-Vision
NemotronForCausalLM	Nemotron-3, Nemotron-4, Minitron	nvidia/Minitron-8B-Base
NemotronNASForCausalLM	NemotronNAS	nvidia/Llama-3_3-Nemotron-Super-49B-v1
Qwen2ForCausalLM	QwQ, Qwen2	Qwen/Qwen2-7B-Instruct
Phi3ForCausalLM	Phi-4	microsoft/Phi-4
Qwen2ForProcessRewardModel	Qwen2-based	Qwen/Qwen2.5-Math-PRM-7B
Qwen2ForRewardModel	Qwen2-based	Qwen/Qwen2.5-Math-RM-72B
Qwen3ForCausalLM	Qwen3	Qwen/Qwen3-8B
Qwen3MoeForCausalLM	Qwen3MoE	Qwen/Qwen3-30B-A3B

Model-Feature Support Matrix(Key Models)

Note: Support for other models may vary. Features marked “N/A” are not applicable to the model architecture.

Model Architecture/Feature	Overlap Scheduler	CUDA Graph	Attention Data Parallelism	Disaggregated Serving	Chunked Prefill	MTP	EAGLE-3(One Model Engine)	EAGLE-3(Two Model Engine)	Torch Sampler	TLLM C++ Sampler	KV Cache Reuse	Sliding Window Attention	Logits Post Processor	Guided Decoding
DeepseekV3ForCausalLM	Yes	Yes	Yes	Yes	Yes [1]	Yes	No	No	Yes	Yes	Yes [2]	N/A	Yes	Yes
Qwen3MoeForCausalLM	Yes	Yes	Yes	Yes	Yes	No	Yes	Yes	Yes	Yes	Yes	N/A	Yes	Yes
Llama4ForConditionalGeneration	Yes	Yes	Yes	Yes	Yes	No	Yes	Yes	Yes	Yes	Untested	N/A	Yes	Yes
GPT-OSS	Yes	Yes	Yes	Yes	No	No	Yes	No	Yes	Yes	No	N/A	Yes	Yes

Multimodal Feature Support Matrix (PyTorch Backend)

Model Architecture/Feature	Overlap Scheduler	CUDA Graph	Chunked Prefill	Torch Sampler	TLLM C++ Sampler	KV Cache Reuse	Logits Post Processor	EPD Disaggregated Serving	Modality
Gemma3ForConditionalGeneration	Yes	Yes	N/A	Yes	Yes	N/A	Yes	No	L + I
HCXVisionForCausalLM	Yes	Yes	No	Yes	Yes	No	Yes	No	L + I
LlavaLlamaModel (VILA)	Yes	Yes	No	Yes	Yes	No	Yes	No	L + I + V
LlavaNextForConditionalGeneration	Yes	Yes	No	Yes	Yes	No	Yes	No	L + I
Llama4ForConditionalGeneration	Yes	Yes	No	Yes	Yes	No	Yes	No	L + I
Mistral3ForConditionalGeneration	Yes	Yes	No	Yes	Yes	No	Yes	No	L + I
Phi4MMForCausalLM	Yes	Yes	No	Yes	Yes	No	Yes	No	L + I + A
Qwen2VLForConditionalGeneration	Yes	Yes	No	Yes	Yes	Yes	Yes	No	L + I + V
Qwen2_5_VLForConditionalGeneration	Yes	Yes	No	Yes	Yes	Yes	Yes	No	L + I + V

Note:

• L: Language

• I: Image

• V: Video

• A: Audio

---

[1]

Chunked Prefill for MLA can only be enabled on SM100.

[2]

KV cache reuse for MLA can only be enabled on SM90/SM100 and in BF16/FP8 KV cache dtype.