Nvidia's Neatron 3.5 Lightning blends Mamba state space models, sparse experts, and limited attention layers for efficient long-context AI performance.
Ask about this video. Answers come from its transcript only — with the timestamp, so you can check them.
Generated from the transcript and can be wrong — check the timestamp.
Key Takeaways
- Neatron 3.5 Lightning is mostly non-transformer, combining Mamba state space models and sparse experts with limited attention.
- Attention layers have high memory costs that Mamba models mitigate by using fixed-size memory updated per token.
- Hybrid architectures with some attention outperform pure state space or pure transformer models on key benchmarks.
- Lightning excels in agentic tasks and long-context reasoning despite fewer active parameters and lower scores on some coding benchmarks.
- The model represents a deliberate design tradeoff for efficiency and capability, distilled from a much larger predecessor.
What the video covers
- Neatron 3.5 Lightning is a 31.12 billion parameter model with only 12% transformer attention layers, the rest being Mamba state space and mixture of experts layers.
- The model runs efficiently on one Nvidia H100 GPU and even on desktop machines at 4-bit precision.
- Attention layers have a high memory cost due to key-value caching that grows with token count, motivating alternatives like Mamba.
- Mamba state space models maintain a fixed-size memory box updated per token, avoiding the growing memory footprint of attention.
- Pure state space models underperform transformers on copying, in-context learning, and long-context reasoning, so a hybrid with some attention layers is used.
- Neatron 3.5 Lightning uses a fixed recipe: 23 Mamba layers, 23 mixture of experts layers, and 6 attention layers spaced evenly.
- Sparse experts provide width sparsity with 128 experts firing selectively per token, while Mamba provides depth sparsity.
- Despite fewer parameters actively used, Lightning matches or exceeds larger models on agentic and long-context tasks but scores lower on some coding benchmarks.
- The architecture reflects a design tradeoff prioritizing efficient long-context reasoning and agentic capabilities over raw parameter count.
- Nvidia distilled Lightning from a much larger 550 billion parameter model and fine-tuned it with reinforcement learning for coding, tool use, and multi-turn tasks.
Full Transcript — Download SRT & Markdown
Speaker A
On the 11th of August, Nvidia published a config file: 52 layers. Six of them are attention. The other 46 are something else entirely, and that ratio is the whole story. The model is Neatron 3.5 Lightning. About 31.12 billion parameters sitting on disk. Nvidia rounds that to 30. 3.12 billion of them wake up for any given token. It runs on one H100. It also runs on a desktop machine at 4-bit today. So it is not, in the strict sense, a transformer. By its own layer count, it is 88% not a transformer. The 12% that still is, that is the part the replace-attention camp expected to disappear. Because for nearly three years, the pitch was that state space models would replace attention outright. That is not what shipped. What shipped is a compromise written down as a fixed recipe. 23 Mamba 2 layers, 23 mixture of experts layers, six attention layers deliberately in that order. Every number here comes from the published config, the model card, or a test somebody else ran. I will name each. So here is the question underneath all of it. If six layers of attention are enough, what were the other 46 doing? Nvidia's launch post, written by Chris Alex and Chintan Patel, does not lead with any of that. It leads with agents and cost. But the architecture is the reason the cost moved, and it is the reason this model exists at this size at all because Lightning was not trained from scratch to be clever. It is distilled down from Neatron 3 Ultra, a 550 billion parameter model, then tuned with reinforcement learning across coding, tool use, and multi-turn work. Attention has a running cost the spec sheets leave out, and it has nothing to do with how smart the model is. When a transformer reads a token, it computes two vectors for it, a key and a value, and it keeps them not for that step, but for the rest of the conversation. That store is the key-value cache. It grows with every single token in every single attention layer, and nothing in it ever gets thrown away. And you can compute exactly what it costs here because Nvidia published the shape: two key-value heads, head dimension 128, 16-bit numbers. That works out to 1 kilobyte per token per attention layer, which sounds like nothing, right up until you multiply it by both of the numbers that matter. If all 52 layers were attention, a million tokens of context would need 52 GB of memory just to remember what it already read. 52 GB before a single new word gets generated. And that is per conversation, not per server. Ten users holding long context means 10 of those side by side on the same card. None of that memory is doing new work. It is what remembering costs. Therefore, the interesting question stopped being how to make attention faster and became whether you need to keep all of it at all. In December 2023, Albert Gu at Carnegie Mellon and Trid Dao at Princeton published an answer, and it came out of control theory. Their idea, called Mamba, replaces the growing list with one fixed box of numbers rewritten as each token arrives. The box stays exactly the same size. Token 1 and token 1 million update exactly the same amount of memory. That is the whole trick, and the name for it is a state space model. Think of the difference this way. Attention keeps the full transcript and can quote any line back on demand. Mamba keeps running notes and has to decide what is worth writing down. In this model, you can size those notes exactly. 64 heads, head dimension 64, state dimension 128 across 23 Mamba layers, 46 megabytes, constant, not 46 at short context and more later, 46 at one token and 46 at a million. So on one side of this model, memory climbs toward gigabytes. And on the other, it is a flat line you could fit on a phone. Which raises the obvious question: if the box stays fixed, why did every lab not simply switch in 2024? Because Nvidia's own researchers went and checked and then published the catch. This is a 2024 paper, and the date matters. They trained 8 billion parameter Mamba, Mamba 2, and transformer models on identical data up to 3 and a half trillion tokens. Same recipe, different architectures, a fair comparison. Pure state space models matched or beat transformers on plenty of tasks. But they fell behind on copying, on in-context learning, and on long context reasoning, which follows directly from the running notes. If a detail did not make it into the notes, you cannot quote it back. Compression is the feature and the flaw at once. And their fix was almost rude in its simplicity. Put a little attention back. Roughly 8% of layers spread evenly through the stack. That hybrid then beat the pure transformer on all 12 standard tasks they ran. Therefore, what Nvidia shipped this month is that 2024 result grown up. 44% Mamba, 44% experts, 11-12% attention. The feed-forward layers got swapped out for sparse experts, and the attention fraction went slightly up, not down. They kept more of it than their own rule of thumb called for. You can read the placement straight out of the file. Attention sits at layers 5, 12, 19, 26, 33, and 42, roughly one every seven. Sebastian Rashka counted the same split independently for the previous model in this family, which is how you know the spacing is deliberate. So, picture it as a relay. Mamba layers compress everything they see into that fixed box cheaply for six layers at a stretch. Then, one attention layer looks back at the real tokens and pulls out the exact detail the compression blurred. Then it hands back, and there is a second kind of sparsity stacked on top of that one. Each mixture layer holds 128 experts, and exactly six of them fire per token plus one shared expert that fires on every token. That shared expert is where the general-purpose knowledge lives. So the six specialists do not each have to relearn it. Sparse in depth from Mamba. Sparse in width from the experts. 30 billion parameters on disk. Three and a half billion doing work. Which brings back the question from the top. Sharper. Now, if six layers of attention can hold a million tokens together, what exactly were the other 46 layers doing in every model you have used until now? The obvious way to answer that is to look at what it scores. Artificial Analysis, who benchmark models independently, put Lightning at 24 on their intelligence index, a composite of nine evaluations spanning agentic work, coding, science, and general knowledge. 24 is the same score as OpenAI's GPT-3 120B, a model with roughly four times the total parameters. That is the headline Nvidia wants you to take away. But QN 3.6 35B scores 32. Meta's Muse Glimmer scores 35. Claude Opus 5 sits at 63. So Lightning loses to models in its own size class, and it is not close. That looks like a bad model, and it is worth spelling out why it is not. Nvidia's own model card publishes the detail with the caveat printed right on it that these are their harness and may differ from other vendors' self-reported figures. SWE Bench verified Lightning 51.56, Quinn 70.12, Terminal Bench 2.1 24.58 against 44.38. Those are not rounding errors. They are a different tier on coding and terminal work. The bigger, longer-thinking model is simply better, and no architecture argument changes it. But look at what sits next to those rows. On Pinchbench and AGENTic suite, Lightning reaches 85.37 against Quinn's 88.07, nearly level at a fraction of the active parameters. And on GDP val, an economically grounded agentic test, Lightning's ELO of 832 beats Neatron 3 Supers 746, a model from its own family at about four times the size. On long context reasoning, it scores 52 against its predecessor's 32, which is the number to watch because long context is precisely where the Mamba half was supposed to hurt it. It did not. So, it loses badly on hard reasoning and beats much larger models on agentic work. That is not a contradiction. It is a design choice, and the mechanism is worth spelling out because this is the question people keep asking about this model. A composite intelligence score rewards two t
Speaker A
parameters sitting on disk. Nvidia rounds that to 30. 3 12 billion of them wake up for any given token. It runs on one H100. It also runs on a desktop machine at 4bit today. So it is not in
Speaker A
the strict sense a transformer. By its own layer count, it is 88% not a transformer. The 12% that still is, that is the part the replace attention camp expected to disappear. Because for nearly 3 years, the pitch was that state
Speaker A
space models would replace attention outright. That is not what shipped. What shipped is a compromise written down as a fixed recipe. 23 Mamba 2 layers, 23 mixture of experts layers, six attention layers deliberately in that order. Every number here comes from the published
Speaker A
config, the model card, or a test somebody else ran. I will name each. So here is the question underneath all of it. If six layers of attention are enough, what were the other 46 doing?
Speaker A
Nvidia's launch post written by Chris Alex and Chintan Patel does not lead with any of that. It leads with agents and cost. But the architecture is the reason the cost moved and it is the reason this model exists at this size at
Speaker A
all because lightning was not trained from scratch to be clever. It is distilled down from neatron 3 ultra a 550 billion parameter model then tuned with reinforcement learning across coding tool use and multi-turn work.
Speaker A
Attention has a running cost the spec sheets leave out and it has nothing to do with how smart the model is. When a transformer reads a token, it computes two vectors for it, a key and a value, and it keeps them not for that step, for
Speaker A
the rest of the conversation. That store is the key value cache. It grows with every single token in every single attention layer, and nothing in it ever gets thrown away. And you can compute exactly what it costs here because
Speaker A
Nvidia published the shape. two key value heads. Head dimension 128, 16- bit numbers. That works out to 1 kilobyte per token per attention layer, which sounds like nothing, right up until you multiply it by both of the numbers that
Speaker A
matter. If all 52 layers were attention, a million tokens of context would need 52 GB of memory just to remember what it already read. 52 GB before a single new word gets generated. And that is per conversation, not per server. 10 users
Speaker A
holding long context means 10 of those side by side on the same card. None of that memory is doing new work. It is what remembering costs. Therefore, the interesting question stopped being how to make attention faster and became
Speaker A
whether you need to keep all of it at all. In December 2023, Albert Gu at Carnegie Melon and Trid Dao at Princeton published an answer and it came out of control theory. Their idea called mamba replaces the growing list with one fixed
Speaker A
box of numbers rewritten as each token arrives. The box stays exactly the same size. Token 1 and token 1 million update exactly the same amount of memory. That is the whole trick and the name for it is a state space model. Think of the
Speaker A
difference this way. Attention keeps the full transcript and can quote any line back on demand. Mamba keeps running notes and has to decide what is worth writing down. In this model, you can size those notes exactly. 64 heads, head
Speaker A
dimension 64, state dimension 128 across 23 mamba layers, 46 megabytes, constant, not 46 at short context and more later, 46 at one token and 46 at a million. So on one side of this model, memory climbs toward gigabytes. And on the other, it
Speaker A
is a flat line you could fit on a phone. Which raises the obvious question, if the box stays fixed, why did every lab not simply switch in 2024? Because Nvidia's own researchers went and checked and then published the catch.
Speaker A
This is a 2024 paper, and the date matters. They trained 8 billion parameter mamba, mamba 2, and transformer models on identical data up to 3 and 1/2 trillion tokens. Same recipe, different architectures, a fair comparison. Pure state space models
Speaker A
matched or beat transformers on plenty of tasks. But they fell behind on copying, on incontext learning, and on long context reasoning, which follows directly from the running notes. If a detail did not make it into the notes, you cannot quote it back. Compression is
Speaker A
the feature and the flaw at once. And their fix was almost rude in its simplicity. Put a little attention back.
Speaker A
Roughly 8% of layers spread evenly through the stack. That hybrid then beat the pure transformer on all 12 standard tasks they ran. Therefore, what Nvidia shipped this month is that 2024 result grown up. 44% mamba, 44% experts, 11 12%
Speaker A
attention. The feed forward layers got swapped out for sparse experts and the attention fraction went slightly up, not down. They kept more of it than their own rule of thumb called for. You can read the placement straight out of the
Speaker A
file. Attention sits at layers 5, 12, 19, 26, 33, and 42, roughly one every seven. Sebastian Rashka counted the same split independently for the previous model in this family, which is how you know the spacing is deliberate. So,
Speaker A
picture it as a relay. Mamba layers compress everything they see into that fixed box cheaply for six layers at a stretch. Then, one attention layer looks back at the real tokens and pulls out the exact detail the compression
Speaker A
blurred. Then it hands back and there is a second kind of sparity stacked on top of that one. Each mixture layer holds 128 experts and exactly six of them fire per token plus one shared expert that fires on every token. That shared expert
Speaker A
is where the general purpose knowledge lives. So the six specialists do not each have to relearn it. Sparse in depth from Mamba. Sparse in width from the experts. 30 billion parameters on disk.
Speaker A
Three and a half billion doing work. Which brings back the question from the top. sharper. Now, if six layers of attention can hold a million tokens together, what exactly were the other 46 layers doing in every model you have
Speaker A
used until now? The obvious way to answer that is to look at what it scores. Artificial analysis, who benchmark models independently, put Lightning at 24 on their intelligence index, a composite of nine evaluations spanning agentic work, coding, science,
Speaker A
and general knowledge. 24 is the same score as OpenAI's GPOS 120B, a model with roughly four times the total parameters. That is the headline Nvidia wants you to take away. But QN 3.635B scores 32. Meta's Muse Glimmer scores
Speaker A
35. Claude Opus 5 sits at 63. So Lightning loses to models in its own size class, and it is not close. That looks like a bad model, and it is worth spelling out why it is not. Nvidia's own
Speaker A
model card publishes the detail with the caveat printed right on it that these are their harness and may differ from other vendors self-reported figures. SWE Bench verified lightning 51.56 Quinn 70.12 terminal bench 2.1 24.58 against 44.38.
Speaker A
Those are not rounding errors. They are a different tier on coding and terminal work. The bigger longer thinking model is simply better and no architecture argument changes it. But look at what sits next to those rows. On pinchbench
Speaker A
and a gentic suite, lightning reaches 85.37 against Quinn's 88.07, nearly level at a fraction of the active parameters. And on GDP val, an economically grounded agentic test, lightning's ELO of 832 beats Nemetron 3 supers 746, a model from its own family
Speaker A
at about four times the size. On long context reasoning, it scores 52 against its predecessors 32, which is the number to watch because long context is precisely where the mamb bea half was supposed to hurt it. It did not. So, it
Speaker A
loses badly on hard reasoning and beats much larger models on agentic work. That is not a contradiction. It is a design choice and the mechanism is worth spelling out because this is the question people keep asking about this
Speaker A
model. A composite intelligence score rewards two things lightning deliberately did not buy. The first is raw knowledge capacity and that scales with total parameters not active ones.
Speaker A
30 billion cannot hold what 120 billion holds. You can see it directly in the rows. On the knowledge benchmark, it scores 17 and 12. On the hardest reasoning set, 11.7. Those are the scores of a small model because it is
Speaker A
one. The second thing is thinking time. Reasoning models climb these indexes by generating more internal tokens per question. Artificial analysis found Lightning used about the same number of output tokens per task as its predecessor while gaining nine points.
Speaker A
So, it did not get smarter by thinking longer. It got smarter per token. That is a different axis and the index does not have a column for it. Artificial analysis said it plainly. It sits behind the most intelligent small models in its
Speaker A
class, but it is built for a different point on the frontier, and that different point is measurable. across the same index. Lightning finishes a task in about half a minute. Quinn 3.635b takes about 3 and a half. GPTOS 12B
Speaker A
about 3.4. Same evaluations, same answers being graded, seven times the wall clock. That gap is what the 24 bought. Whether it was worth buying depends entirely on what your agent does all day. Therefore, the claim Nvidia leads with is speed. Up to four times
Speaker A
the output of similar size models. That is a vendor number so it deserves an audit. And part of it is the quantization. This part is unusual.
Speaker A
Nvidia did not just release 4-bit weights alongside the 16- bit ones. They pre-trained the model in 4bit across more than 20 trillion tokens using a format called NVFP4.
Speaker A
NVFP4 splits a tensor into blocks of 16 values, gives each block its own 8-bit scale, then applies a second scale across the whole tensor. Small blocks mean the scale tracks local detail instead of averaging it away. That works
Speaker A
out to about 4 and a half bits per value, roughly three and a half times smaller than 16 bit. And artificial analysis measured the 4-bit version at 24 as well. Same score, a quarter of the memory, which is why this fits on a
Speaker A
desk. Through a llama, it is a 25 GB download sitting at 26 GB of memory with a quarter of a million tokens of context loaded. So Siam Pac, who runs Cube Simplify, pulled it onto a DGX Spark the
Speaker A
hour it landed and measured it himself rather than quoting the charts. The first thing he hit was mundane. Alama refused the download with an HTTP412 because support for the architecture landed in a version released hours after the weights did. Once running, he
Speaker A
measured about 72 tokens a second on pros, 84 to 87 on JSON and summaries through VLLM with the 4-bit weights and the DSpark drafter. 108 prefill roughly doubled too to around 5,400 tokens a second. Then he did the thing vendors do
Speaker A
not do. He ran a rival on the identical box, same runtime, same prompt. QN 3.535B managed 78 tokens a second slightly ahead of lightning. His conclusion, in his words, the four times figure comes from hosted endpoints running the full
Speaker A
draft model stack and you do not see it through Alama today against its own big brother Nematron 3 Super. It is a real three and a bit times which is the accurate version of the claim and still a real number. But he also found the one
Speaker A
that matters more for an agent. Same prompt, same correct tool call, both models right. Lightning got there in 485 tokens in 6 seconds. QN took 1,953 tokens and 26 seconds. He is careful to say one prompt is not a benchmark and he
Speaker A
is right, but it points the same way as the index timing from a completely different measurement. And you can control that dial directly. Reasoning switches off per request or gets capped with a token budget, which is the actual
Speaker A
product idea, not a smarter model, but a model whose cost per step you set.
Speaker A
Because an agent makes thousands of cheap calls and a handful of hard ones. And until now, you paid frontier prices for all of them. So, Nvidia shipped a second thing the same day. Nemo Switchyard, an open- source router that
Speaker A
sends each step to whichever model should handle it. Plans go up to the Frontier model. Execution comes down to Lightning. The interesting part is that somebody outside Nvidia measured it.
Speaker A
Lang Chain ran 145 multi-turn agent tasks through it, routing between Lightning and Claude Opus 4.8 8 and published the receipts. Here is the split that matters. Lightning handled 93% of the calls for 10.4% of the spend.
Speaker A
Opus handled 7% of the calls and carried 68% of the bill. Per call, Opus cost about 87 times more. Total cost fell 74% and accuracy fell six points from 86 to 80. Ramp reported a similar shape on their own internal benchmark, 58%
Speaker A
cheaper, 33% faster. But Langchain also published the part A marketing page cuts. Routing scored 2.3 points above simply using the cheap model for everything and their own runs vary by 2.7 points. So on their workload they could not prove the router beat just
Speaker A
using lightning for the lot and they said so in writing. Their framing is the one to keep. Routing is the cost of not having to guess which requests are hard.
Speaker A
Step back because this is bigger than one model. Every serious lab shipped this same compromise this year and every one of them picked a different fraction.
Speaker A
QN keeps one attention layer in four. Kimmy linear roughly the same. Ling 2.5 keeps 1 and 8. Nvidia keeps one and nine here and one in 11 on their larger model. Same idea, five different answers. Meanwhile, Mamba 3 landed at
Speaker A
ICLR this year from GU and DAO again matching Mamba 2's quality at about half the state size. and one line to keep two things apart because they get blurred constantly. World models like Nvidia's own Cosmos are a separate research lane
Speaker A
about predicting physical reality. Different bet, different scoreboard, not something you route agent traffic through. So, the verdict, the Transformer versus Mamba argument is over and neither side won it. The ratio won and it is no longer a research
Speaker A
position. It is a manufacturing decision at the company that sells the GPUs, which is what actually changed this month. For the 95% of you running agents that make many cheap calls, this is the better buy. At 5 cents in and 20 cents
Speaker A
out per million tokens, for the 5% doing one hard problem at a time, that index score is telling you the truth. I would still take Quinn 3.635b 635B for a coding agent because 70 against 51 on SWEBench is the entire job and the
Speaker A
villain here is not a company. It is a habit. One composite number priced this model as mediocre while the thing it is good at has no column on that scoreboard which leaves the open question. Four labs, four different fractions. And as
Speaker A
Sebastian Rashka points out, without matched ablations, no lab can prove the ratio is the thing that did it. So if that one number is the whole ball game, why is every lab still guessing at it?
Topics:Neatron 3.5 LightningNvidia AI modelMamba state space modelMixture of ExpertsTransformer alternativesLong context reasoningSparse expertsAttention layersAI architectureReinforcement learning tuning











