算力商业
developer.nvidia.com ·
@Elizabeth Goodman
前沿推理到达边缘:如何在NVIDIA Jetson上部署和优化模型
NVIDIA技术博客讨论在边缘设备上运行推理和代理AI的挑战,指出多步推理模型过去因体积过大而难以在边缘运行,现提供部署和优化方法。
AI推荐理由 NVIDIA技术博客介绍如何在Jetson边缘设备上部署和优化推理模型,解决边缘运行多步推理模型的难题,对边缘AI部署有实际指导意义。
80
正文
边缘计算
前沿推理到达边缘:如何在NVIDIA Jetson上部署和优化模型
2026年9月4日
作者:Khalil BenKhaled、Aditya Sahu和Chitoku Yato
喜欢
L
T
F
R
E
AI生成摘要
2026年发布的紧凑型开放模型现在提供了以前需要大型数据中心系统的推理和智能体能力,而NVIDIA Jetson可以在边缘本地运行它们。
Nemotron 3.5 Lightning采用混合专家架构,总参数为300亿,但每个令牌仅激活30亿,而Qwen3.8-27B是一个密集模型,激活全部270亿参数,使它们各自适合不同的智能体工作负载。
NVFP4量化减少了模型操作所需的工作量和内存,而投机解码可以在每个验证步骤中生成多个被接受的令牌;它们一起在Jetson上比BF16提供了高达6.28倍的解码吞吐量加速。
最快的投机解码配置因模型而异:Nemotron 3.5 Lightning使用DSpark表现最佳,而Qwen3.8-27B使用DFlash2表现最佳,因此开发人员应使用目标模型测试方法和草稿检查点。
使用代表性提示进行应用级验证至关重要,因为吞吐量因工作负载类别而异,通用基准无法确认检查点是否保留了特定数据的关键行为。
后续步骤
访问Jetson AI Lab模型页面,探索模型、基准结果、推荐配方以及跨Jetson平台的性能比较。
查看在Jetson上运行LLM和VLM的教程,获取动手部署指导。
探索投机解码入门教程,在流行框架上实现优化技术。
由NVIDIA Nemotron提供支持。AI生成的内容可能不完整地总结信息。验证重要信息。了解更多
在边缘运行推理和智能体AI一直比应有的难度更大。直到最近,能够进行多步推理的模型太大,无法在边缘硬件上本地运行。构建智能体的开发人员不得不通过数据中心路由推理,增加了网络依赖,提高了成本,并暴露了可能需要在设备上保留的数据。
这一限制正在解除。整个夏天发布的几个模型家族共同标志着边缘AI的转折点。这一代新的紧凑型开放模型现在提供了仅几个月前需要大型数据中心系统的推理和智能体能力,而NVIDIA Jetson今天就可以运行它们。
这可能为驾驶室助手、实时异常检测以及在恶劣或偏远环境中工作的机器人提供动力。现场专家可以减少故障排除时间,关键系统可以在连接受限或不可用时继续运行。
本文介绍了使用Nemotron 3.5 Lightning和Qwen3.8-27B作为示例,在Jetson上部署这一代新开放模型所需了解的内容。您将学习在比较模型架构时要注意什么,如何应用推理优化技术以充分利用硬件,以及如何验证工作负载的配置。
具体来说,本文回答了以下开发人员问题:
如何为Jetson选择推理模型?
NVFP4量化和投机解码如何提高推理性能?
如何使用vLLM服务Nemotron 3.5 Lightning和Qwen3.8-27B?
如何验证应用程序的配置?
下面的图1显示了这一转变。它按模型大小和发布日期绘制了Artificial Analysis Intelligence Index。2026年发布的开放模型现在达到了与2025年领先模型相似的分数,同时使用的参数少得多。
图1. 支持边缘的2026年模型(绿色)以一小部分参数数量达到了与2025年前沿模型相当的智能分数
如何为Jetson选择推理模型?
更好的训练方法和更高效的架构正在推动这一变化。例如,蒸馏将Nemotron 3 Ultra的一些能力转移到较小的Nemotron 3.5 Lightning模型中。不同的架构也在能力、内存使用和生成速度方面产生了不同的权衡。
Qwen3.8-27B是一个密集模型,因此每个令牌激活全部270亿参数。Nemotron 3.5 Lightning使用混合专家(MoE)架构。它有300亿总参数,但每个令牌仅激活30亿。这使得这两个模型适合不同类型的工作负载。
这些差异对于长时间运行的智能体尤其重要。例如,智能体可以使用实时传感器数据和设备日志监控系统,采取经批准的正确行动,根据预定义测试验证结果,并且仅在需要时升级给专家。所有这些都在边缘本地运行,无需互联网连接,保持低延迟。
Nemotron 3.5 Lightning非常适合这些响应密集型工作流,其中更快的令牌生成可以缩短整体过程。Qwen3.8-27B更适合需要更少、更困难决策并允许智能体花更多时间生成每个响应的任务。
在选择之前,在应用程序所需的决策、工具和响应模式上对两个模型进行基准测试。
在Jetson上,这些智能体循环可以在它们所工作的传感器和系统旁边运行。您可以通过流行框架(包括vLLM和llama.cpp)在本地部署模型,因此推理循环并不完全依赖数据中心。
Gemma 4 E4B是Jetson Orin Nano的坚实起点。对于Jetson AGX Orin和Jetson AGX Thor,Nemotron 3.5 Lightning和Qwen3.8-27B是强有力的选择。这些模型家族在流行推理引擎中具有高质量的量化检查点和优化的部署选项。
如何优化Jetson上的推理模型推理?
两种互补技术可以提高推理性能:NVFP4量化
减少了模型操作所需的工作量和内存,而投机解码可以在每个验证步骤中生成多个被接受的令牌。
模型架构设定了起点,但服务选择也会影响性能。下面的图2比较了两种模型在BF16、NVFP4以及NVFP4加上我们为每个模型测试的最快投机解码配置下的表现。
图2. NVFP4量化和投机解码一起在Jetson上比BF16提供了高达6.28倍的解码吞吐量加速
如上图2所示,我们一次添加两个优化。BF16提供了基线。NVFP4添加了量化。最终配置将NVFP4与我们为每个模型测试的最快投机解码配置相结合。
在解码过程中,模型一次生成一个令牌的响应。每个令牌通常需要再次通过模型。这为您提供了两种提高性能的方法:减少每次通过的工作量或从每次通过中产生更多令牌。
量化采用第一种方法。较低精度的值减少了GPU在每次通过期间必须移动和处理的数据量。使用NVFP4等格式,您可以提高生成速度并减少内存使用,同时保持接近BF16的质量。
投机解码采用第二种方法。较小的草稿模型提出几个令牌,主模型一起验证它们。主模型仍然做出最终决定。如果它接受几个提议的令牌,则生成在一个验证步骤中前进几个令牌。
下面的视频1显示了有和没有投机解码的响应生成,说明了由此产生的加速。
视频1. 在NVIDIA Jetson上比较有和没有投机解码的Qwen3.5 9B NVFP4推理
有几种方法可以产生这些草稿,包括MTP、DFlash和DSpark。这三种方法都可以在Jetson上运行,但它们生成和评估提议的方式不同。我们测试了可用的方法和草稿检查点,而不是假设一种配置对每个模型都表现最佳。
这两种优化相辅相成。NVFP4降低了每次通过的成本,而投机解码增加了每次通过产生的被接受令牌数量。它们一起比单独任何一种优化都能提高更多性能。
最快的投机解码配置在两种模型之间有所不同。Nemotron 3.5 Lightning使用DSpark表现最佳,而Qwen3.8-27B使用DFlash2表现最佳。使用您计划部署的模型测试方法和草稿检查点,而不是假设一种配置对每个模型都最佳。
先决条件
在运行以下命令之前,请确认您具备:
Jetson AGX Thor或Jetson AGX Orin
配置了NVIDIA Container Runtime和Docker的JetPack 7.2。
足够的存储空间用于模型和草稿检查点
已接受NVIDIA Nemotron和Qwen3.8检查点的许可条款
对于Nemotron 3.5 Lightning,您可以在Jetson AGX Thor或Jetson AGX Orin上运行我们测试的最快配置,该配置将NVFP4与DSpark相结合。
首先,启动vllm/vllm-openai:v0.28.0容器:
docker run --pull=always --runtime nvidia --rm -it \
--network host \
--ipc=host \
-v ~/.cache/huggingface:/root/.cache/huggingface \
--entrypoint bash \
vllm/vllm-openai:v0.28.0
然后,在容器内运行以下命令:
vllm serve nvidia/NVIDIA-Nemotron-3.5-Lightning-30B-A3B-NVFP4 \
--reasoning-parser nemotron_v3 \
--enable-auto-tool-choice \
--tool-call-parser qwen3_coder \
--max-model-len 128000 \
--kv-cache-dtype fp8 \
--gpu-memory-utilization 0.7 \
--trust-remote-code \
--max-num-batched-tokens 16384 \
--enable-prefix-caching \
--speculative-config '{"method":"dspark","model":"nvidia/NVIDIA-Nemotron-3.5-Lightning-30B-A3B-NVFP4-DSpark","num_speculative_tokens":5}' \
--mamba-backend flashinfer \
--mamba-ssm-cache-dtype float16 \
--enable-mamba-cache-stochastic-rounding \
--mamba-cache-philox-rounds 5 \
--mamba-cache-mode align
对于Qwen3.8-27B,您可以使用上面命令启动的相同容器,并在Jetson AGX Thor或Jetson AGX Orin上使用以下命令运行我们测试的最快配置,该配置将NVFP4与DFlash2相结合:
VLLM_GDN_DECODE_KERNEL=triton vllm serve Inferact/Qwen3.8-27B-NVFP4 \
--served-model-name qwen38 \
--reasoning-parser qwen3 \
--enable-auto-tool-choice \
--tool-call-parser qwen3_coder \
--max-model-len 50000 \
--max-num-seqs 8 \
--gpu-memory-utilization 0.85 \
--trust-remote-code \
--speculative-config '{"method":"dflash","model":"incoai/Qwen3.8-27B-DFlash2","num_speculative_tokens":7}'
每种方法创建和评估草稿令牌的方式不同,这会影响其提议的成本和准确性。
MTP使用与主模型一起训练的预测头来提议几个未来令牌。您现在可以将MTP与许多主流模型家族一起使用,包括Qwen、Gemma和Nemotron。
DFlash使用单独的基于扩散的草稿模型并行提议一个令牌块。它目前支持最广泛的兼容草稿检查点选择。DSpark在DFlash的基础上通过纠正草稿并提前停止弱提议来构建。当匹配的检查点可用时,DSpark可以更快,但它支持的检查点更少。
使用代表性工作负载验证性能
模型级基准可以帮助您识别强大的投机解码配置。然而,应用程序生成不同类型的文本,性能可能随工作负载而变化。为了测量这种影响,我们保持每个模型的最快配置固定,并测试了四个SpeedBench类别:写作、推理、摘要和检索增强生成。
图3. 投机解码加速因工作负载而异,RAG和写作对两种模型受益最大
在我们测试的类别中,相同的方法对每个模型仍然是最快的,但吞吐量仍然因工作负载而异。Nemotron 3.5 Lightning与DSpark
的输出令牌速率在123.01到138.02之间,而Qwen3.8-27B与DFlash2的输出令牌速率在27.69到34.44之间。
使用代表预期应用程序的提示来验证您的投机解码配置。代表性数据集将帮助您选择最适合您的模型和用例的方法和草稿检查点。
何时应该训练自定义检查点?
对于大多数应用程序,从现有的量化检查点和草稿模型开始。这通常足以获得强大的准确性和有用的加速,而无需自己训练任何东西。在部署配置之前,使用来自应用程序的提示进行测试。通用基准无法告诉您检查点是否保留了与您的数据相关的行为。
如果量化降低了准确性,请使用NVIDIA Model Optimizer通过量化感知训练或蒸馏对量化模型进行微调。量化感知训练在训练期间模拟较低精度。量化感知蒸馏还使用更高精度的教师模型来帮助量化模型保留原始模型的行为。
这种额外的调整对于小的准确性变化很重要的专业工作负载最有用。要了解如何使用任一方法训练量化模型,请遵循NVIDIA Model Optimizer QAT和QAD教程。
对投机解码应用相同的方法。公共草稿检查点可能与您的模型兼容,但仍提供比预期更少的加速。加速取决于主模型接受提议令牌的频率。当接受率低时,产生和验证草稿的成本可能会减少收益。如果发生这种情况,请遵循vLLM Speculators训练指南,使用代表性应用程序数据训练兼容的投机器。Speculators支持包括MTP、EAGLE-3、DFlash和DSpark在内的方法。训练后,在您自己的提示上测量草稿令牌接受率和解码吞吐量。
大多数应用程序不需要自定义训练。从可用的检查点开始,测量它们的准确性和性能,并且仅当这些结果显示明显差距时才训练您自己的。
开始使用
Jetson支持最新的开放模型,具有优化的运行时、量化检查点和投机解码。有了这个基础,您就可以从测试模型转向构建和交付边缘应用程序。
要了解有关这些模型的更多信息,了解如何对它们进行基准测试,找到推荐的配方,并比较跨Jetson平台的性能,请访问Jetson AI Lab模型页面。
有关更多动手指导,请参阅我们在Jetson上运行LLM和VLM、对生成式AI模型进行基准测试以及在流行框架上开始使用投机解码的教程。
喜欢
标签
智能体AI / 生成式AI | 边缘计算 | 通用 | Jetson | 中级技术 | 教程 | JetPack | Jetson Orin | 物理AI | Thor
关于作者
关于Khalil BenKhaled
Khalil BenKhaled是NVIDIA Jetson团队的技术营销工程师。凭借创建和扩展AI和机器人解决方案的背景,Khalil喜欢让边缘AI对开发人员和工程师来说从根本上更简单、更易访问。
查看Khalil BenKhaled的所有帖子
关于Aditya Sahu
Aditya Sahu是NVIDIA Jetson Edge AI平台团队的技术营销工程师。他负责推动平台采用并展示其技术能力。他积极与NVIDIA开发者社区互动,倡导在Jetson上进行AI模型推理和SDK。他之前在Synaptics工作,专注于构建Edge AI解决方案,并在Nuvoton和Netcracker担任ML和软件工程职位。他拥有圣何塞州立大学人工智能硕士学位和Siddaganga理工学院计算机科学学士学位。
查看Aditya Sahu的所有帖子
关于Chitoku Yato
Chitoku Yato是节能、紧凑且可扩展的NVIDIA Jetson Edge AI平台的技术产品营销经理,负责确保平台上的最佳开发者体验。他还与充满活力的NVIDIA开发者社区密切合作,帮助推广在Jetson上使用预训练AI模型、开发者SDK和对云原生技术的支持,以帮助客户构建、部署和管理基于AI的自主机器。
查看Chitoku Yato的所有帖子
查看原文(中文正文为 AI 翻译)
Edge Computing
Frontier Reasoning Reaches the Edge: How to Deploy and Optimize Models on NVIDIA Jetson
Sep 04, 2026
By Khalil BenKhaled, Aditya Sahu and Chitoku Yato
Like
L
T
F
R
E
AI-Generated Summary
Compact open models released in 2026 now deliver reasoning and agentic capabilities that previously required large data center systems, and NVIDIA Jetson can run them locally at the edge.
Nemotron 3.5 Lightning uses a mixture-of-experts architecture with 30 billion total parameters but activates only 3 billion per token, while Qwen3.8-27B is a dense model activating all 27 billion parameters, making each suited to different agent workloads.
NVFP4 quantization reduces the work and memory required for model operations, and speculative decoding generates multiple accepted tokens per verification step; together they deliver up to 6.28x decode throughput speedup over BF16 on Jetson.
The fastest speculative decoding configuration differs by model: Nemotron 3.5 Lightning performs best with DSpark, while Qwen3.8-27B performs best with DFlash2, so developers should test methods and draft checkpoints with their target model.
Application-level validation with representative prompts is essential because throughput varies by workload category, and general benchmarks cannot confirm whether a checkpoint preserves behavior critical to specific data.
Next Steps
Visit the Jetson AI Lab Models page to explore models, benchmark results, recommended recipes, and performance comparisons across Jetson platforms.
Review the running LLMs and VLMs on Jetson tutorial for hands-on deployment guidance.
Explore the getting started with speculative decoding tutorial to implement optimization techniques on popular frameworks.
Powered by NVIDIA Nemotron. AI-generated content may summarize information incompletely. Verify important information. Learn more
Running reasoning and agentic AI at the edge has been harder than it needs to be. Until recently, models capable of multi-step reasoning were too large to run locally on edge hardware. Developers building agents have had to route inference through a data center, adding network dependency, increasing costs, and exposing data that may need to stay on device.
That constraint is lifting. Several model families released throughout the summer have collectively marked a turning point for edge AI. This new generation of compact open models now delivers reasoning and agentic capabilities that required large data center systems only a few months ago, and NVIDIA Jetson can run them today.
This could power in-cab assistants, real-time anomaly detection, and robots working in harsh or remote environments. On-site experts could spend less time troubleshooting, and critical systems could keep running when connectivity is limited or unavailable.
This post walks through what you need to know to deploy this new generation of open models on Jetson, using Nemotron 3.5 Lightning and Qwen3.8-27B as examples. You’ll learn what to look for when comparing model architectures, how to apply inference optimization techniques to get the most out of your hardware, and how to validate the configuration for your workload.
Specifically, this post answers the following developer questions:
How do you choose a reasoning model for Jetson?
How do NVFP4 quantization and speculative decoding improve inference performance?
How do you serve Nemotron 3.5 Lightning and Qwen3.8-27B with vLLM?
How do you validate the configuration for your application?
Figure 1, below, shows this shift. It plots the Artificial Analysis Intelligence Index by model size and release date. Open models released in 2026 now reach scores similar to leading models from 2025, while using far fewer parameters.
Figure 1. Edge-capable 2026 models (green) reach intelligence scores comparable to 2025 frontier models at a fraction of the parameter count
How do you choose a reasoning model for Jetson?
Better training methods and more efficient architectures are driving this change. For example, distillation transfers some of Nemotron 3 Ultra’s capabilities into the smaller Nemotron 3.5 Lightning model. Different architectures also create different tradeoffs in capability, memory use, and generation speed.
Qwen3.8-27B is a dense model, so it activates all 27 billion parameters for each token. Nemotron 3.5 Lightning uses a mixture-of-experts (MoE) architecture. It has 30 billion total parameters but activates only 3 billion for each token. This makes the two models suitable for different types of workloads.
These differences become especially important for long-running agents. For example, an agent can monitor a system using live sensor data and device logs, take approved corrective actions, verify the results against predefined tests, and escalate to an expert only when needed. All of this runs locally at the edge without an internet connection, keeping latency low.
Nemotron 3.5 Lightning is a good fit for these response-heavy workflows, where faster token generation can shorten the overall process. Qwen3.8-27B is a better fit for tasks that require fewer, harder decisions and allow the agent to spend more time generating each response.
Benchmark both models on the decisions, tools, and response patterns required by your application before selecting one.
On Jetson, these agent loops can run next to the sensors and systems they work with. You can deploy models locally through popular frameworks, including vLLM and llama.cpp, so the reasoning loop does not depend entirely on the data center.
Gemma 4 E4B is a strong starting point for Jetson Orin Nano. For Jetson AGX Orin and Jetson AGX Thor, Nemotron 3.5 Lightning and Qwen3.8-27B are strong options. These model families have high-quality quantized checkpoints and optimized deployment options across popular inference engines.
How do you optimize reasoning-model inference on Jetson?
Two complementary techniques can improve inference performance: NVFP4 quantization reduces the work and memory required for model operations, while speculative decoding can generate multiple accepted tokens per verification step.
Model architecture sets the starting point, but serving choices also affect performance. Figure 2, below, compares both models in BF16, NVFP4, and NVFP4 with the fastest speculative decoding configuration we tested for each model.
Figure 2. NVFP4 quantization and speculative decoding together deliver up to 6.28x decode throughput speedup over BF16 on Jetson
As shown in Figure 2, above, we add two optimizations, one at a time. BF16 provides the baseline. NVFP4 adds quantization. The final configuration combines NVFP4 with the fastest speculative decoding configuration we tested for each model.
During decoding, a model generates its response one token at a time. Each token usually requires another pass through the model. This gives you two ways to improve performance: reduce the work in each pass or produce more tokens from each pass.
Quantization takes the first approach. Lower-precision values reduce the amount of data the GPU must move and process during each pass. With a format such as NVFP4, you can improve generation speed and reduce memory use while keeping quality close to BF16.
Speculative decoding takes the second approach. A smaller draft model proposes several tokens, and the main model verifies them together. The main model still makes the final decision. If it accepts several proposed tokens, generation advances by several tokens in one verification step.
Video 1, below, shows a response generation with and without speculative decoding, illustrating the resulting speedup.
Video 1. Comparing Qwen3.5 9B NVFP4 inference with and without speculative decoding on NVIDIA Jetson
Several methods can produce these drafts, including MTP, DFlash, and DSpark. All three run on Jetson, but they generate and evaluate proposals differently. We tested the available methods and draft checkpoints instead of assuming that one configuration would perform best for every model.
The two optimizations complement each other. NVFP4 reduces the cost of each pass, while speculative decoding increases the number of accepted tokens produced from each pass. Together, they improve performance more than either optimization alone.
The fastest speculative decoding configuration differed between the two models. Nemotron 3.5 Lightning performed best with DSpark, while Qwen3.8-27B performed best with DFlash2. Test the method and draft checkpoint with the model you plan to deploy instead of assuming that one configuration will work best for every model.
Prerequisites
Before running the following commands, confirm that you have:
A Jetson AGX Thor or Jetson AGX Orin
JetPack 7.2 with NVIDIA Container Runtime and Docker configured.
Enough storage for the model and draft checkpoints
Accepted the license terms for NVIDIA Nemotron and Qwen3.8 checkpoints
For Nemotron 3.5 Lightning, you can run the fastest configuration we tested, which combines NVFP4 with DSpark, on Jetson AGX Thor or Jetson AGX Orin.
First, start the vllm/vllm-openai:v0.28.0 container:
docker run --pull=always --runtime nvidia --rm -it \
--network host \
--ipc=host \
-v ~/.cache/huggingface:/root/.cache/huggingface \
--entrypoint bash \
vllm/vllm-openai:v0.28.0
Then, run the following command inside the container:
vllm serve nvidia/NVIDIA-Nemotron-3.5-Lightning-30B-A3B-NVFP4 \
--reasoning-parser nemotron_v3 \
--enable-auto-tool-choice \
--tool-call-parser qwen3_coder \
--max-model-len 128000 \
--kv-cache-dtype fp8 \
--gpu-memory-utilization 0.7 \
--trust-remote-code \
--max-num-batched-tokens 16384 \
--enable-prefix-caching \
--speculative-config '{"method":"dspark","model":"nvidia/NVIDIA-Nemotron-3.5-Lightning-30B-A3B-NVFP4-DSpark","num_speculative_tokens":5}' \
--mamba-backend flashinfer \
--mamba-ssm-cache-dtype float16 \
--enable-mamba-cache-stochastic-rounding \
--mamba-cache-philox-rounds 5 \
--mamba-cache-mode align
For Qwen3.8-27B, you can use the same container launched with the command above and run the fastest configuration we tested, which combines NVFP4 with DFlash2, on Jetson AGX Thor or Jetson AGX Orin using the following command:
VLLM_GDN_DECODE_KERNEL=triton vllm serve Inferact/Qwen3.8-27B-NVFP4 \
--served-model-name qwen38 \
--reasoning-parser qwen3 \
--enable-auto-tool-choice \
--tool-call-parser qwen3_coder \
--max-model-len 50000 \
--max-num-seqs 8 \
--gpu-memory-utilization 0.85 \
--trust-remote-code \
--speculative-config '{"method":"dflash","model":"incoai/Qwen3.8-27B-DFlash2","num_speculative_tokens":7}'
Each method creates and evaluates draft tokens differently, which affects the cost and accuracy of its proposals.
MTP uses prediction heads trained with the main model to propose several future tokens. You can now use MTP with many mainstream model families, including Qwen, Gemma, and Nemotron.
DFlash uses a separate diffusion-based draft model to propose a block of tokens in parallel. It currently supports the widest selection of compatible draft checkpoints. DSpark builds on DFlash by correcting drafts and stopping weak proposals early. DSpark can be faster when a matching checkpoint is available, but it supports fewer checkpoints.
Validate performance with representative workloads
A model-level benchmark can help you identify a strong speculative decoding configuration. However, applications generate different kinds of text, and performance can change with the workload. To measure this effect, we kept the fastest configuration for each model fixed and tested four SpeedBench categories: writing, reasoning, summarization, and retrieval-augmented generation.
Figure 3. Speculative decoding speedup varied by workload, with RAG and writing benefiting most for both models
The same method remained the fastest for each model across the categories we tested, but throughput still varied by workload. Nemotron 3.5 Lightning with DSpark ranged from 123.01 to 138.02 output tokens/s, while Qwen3.8-27B with DFlash2 ranged from 27.69 to 34.44 output tokens/s.
Validate your speculative decoding configuration with prompts that represent the intended application. A representative dataset will help you choose the method and draft checkpoint that work best for your model and use case.
When should you train a custom checkpoint?
For most applications, start with an existing quantized checkpoint and draft model. This is usually enough to get strong accuracy and a useful speedup without training anything yourself. Before deploying the configuration, test it with prompts from your application. General benchmarks cannot tell you whether a checkpoint preserves the behavior that matters for your data.
If quantization reduces accuracy, use NVIDIA Model Optimizer to fine-tune the quantized model with quantization-aware training or distillation. Quantization-aware training simulates lower precision during training. Quantization-aware distillation also uses a higher-precision teacher model to help the quantized model retain the original model’s behavior.
This additional tuning is most useful for specialized workloads where small accuracy changes matter. To learn how to train a quantized model with either approach, follow the NVIDIA Model Optimizer QAT and QAD tutorial.
Apply the same approach to speculative decoding. A public draft checkpoint can be compatible with your model but still provide less acceleration than expected. The speedup depends on how often the main model accepts the proposed tokens. When the acceptance rate is low, the cost of producing and verifying drafts can reduce the benefit. If this happens, follow the vLLM Speculators training guide to train a compatible speculator with representative application data. Speculators supports methods including MTP, EAGLE-3, DFlash, and DSpark. After training, measure both the draft-token acceptance rate and decode throughput on your own prompts.
Most applications will not need custom training. Start with the available checkpoints, measure their accuracy and performance, and train your own only when those results show a clear gap.
Get started
Jetson supports the latest open models with optimized runtimes, quantized checkpoints, and speculative decoding. With this foundation in place, you can move from testing models to building and shipping edge applications.
To learn more about these models, see how to benchmark them, find recommended recipes, and compare performance across Jetson platforms, visit the Jetson AI Lab Models page.
For more hands-on guidance, see our tutorials on running LLMs and VLMs on Jetson, benchmarking generative AI models, and getting started with speculative decoding on popular frameworks.
Like
Tags
Agentic AI / Generative AI | Edge Computing | General | Jetson | Intermediate Technical | Tutorial | JetPack | Jetson Orin | Physical AI | Thor
About the Authors
About Khalil BenKhaled
Khalil BenKhaled is a technical marketing engineer on the Jetson team at NVIDIA. With a background in creating and scaling AI and robotics solutions, Khalil enjoys making AI at the edge fundamentally simpler and more accessible for developers and engineers.
View all posts by Khalil BenKhaled
About Aditya Sahu
Aditya Sahu is a technical marketing engineer on the NVIDIA Jetson Edge AI Platform team. He is responsible for driving the adoption and showcasing the technical capabilities of the platform. He actively engages with the NVIDIA developer community, advocating for AI model inference and SDKs on Jetson. He previously worked at Synaptics, focusing on building Edge AI solutions, and held ML and software engineering roles at Nuvoton and Netcracker. He holds an MS in Artificial Intelligence from San José State University and a BS in Computer Science from Siddaganga Institute of Technology.
View all posts by Aditya Sahu
About Chitoku Yato
Chitoku Yato is technical product marketing manager of the power-efficient, compact and scalable NVIDIA Jetson Edge AI platform, where he is responsible for ensuring the best developer experience on the platform. He also works closely with the vibrant NVIDIA developer community to help evangelize the use of pre-trained AI models, developer SDKs and support for cloud-native technologies on Jetson to help customers build, deploy, and manage AI-based autonomous machines.
View all posts by Chitoku Yato