Ziteng Sun retweeted
Excited to share last summer's work at Google Research!
Most hybrid models today are static: each token sees the same interleaved pattern of your favorite linear model and attention.
Oryx instead varies the model used across the sequence through shared representations.
1/
⚡ Excited to announce Gemini 3.1 Flash-Lite! We’ve set a new standard for efficiency and capability to give developers our fastest, most cost-effective Gemini 3 model yet.
We engineered this model with thinking levels, allowing it to handle high-volume queries instantly, while scaling up its reasoning for complex edge cases.
By the numbers:
⏱️ 2.5X faster time-to-first-token than 2.5 Flash while being significantly higher quality
📉 $0.25 per 1M input tokens
📊 1432 Elo on LMArena & 86.9% on GPQA Diamond
Thrilled to see what developers build with this kind of speed and quality at scale. Available now in Google AI Studio and Vertex AI.
blog.google/innovation-and-a…
Ziteng Sun retweeted
Introducing SPECS (SPECulative test time Scaling), a test-time scaling (TTS) algorithm with pareto-frontier latency/accuracy trade-off.
Scaling test-time compute improves LLM reasoning but imposes a latency overhead.
Prior work optimizes TTS accuracy as a function of FLOPS, we propose to further reduce latency by addressing the memory bottleneck of LLM inference through speculative drafts. See a breakdown of the method below.
(1/n) 🧵 👇
Ziteng Sun retweeted
I'm hiring a student researcher to work on RL and RLM-flavored things. DM me if interested
Google Student Researcher Program 2026 is now OPEN!
Work on REAL AI/ML projects with:
• Google Research
• DeepMind
• Google Cloud
Open to: Bachelors / Masters / PhD
Duration: 3–12 months
Deadline: March 31
If you're serious about AI, this is your shot.
Apply here
google.com/about/careers/app…
I will be at NeurIPS from today to Dec. 7th. Excited to meet old and new friends at the conference. Happy to chat about anything related to LLM efficiency, RL, and differential privacy. #NeurIPS2025
At the Wednesday noon session (11 AM – 2PM), I will be presenting our spotlight work: Private Set Union with Multiple Contributions (#1314), where we establish fundamental limits on the utility of discovering set unions privately, and how we can bypass the limit by leveraging a prediction.
Joint work with awesome collaborators at Google Research: Travis Dick, Haim Kaplan, Alex Kulesza, Uri Stemmer, and @th33rtha.
Ziteng Sun retweeted
This is Gemini 3: our most intelligent model that helps you learn, build and plan anything.
It comes with state-of-the-art reasoning capabilities, world-leading multimodal understanding, and enables new agentic coding experiences. 🧵
Ziteng Sun retweeted
The main ingredient that led to GRPO's performance leap is the calibration of the reward/value via multiple rollouts per prompt.
Let me elaborate on what I mean by that and a cheaper way of doing it offline.
Ziteng Sun retweeted
Happening now at poster E-2804.
Come talk to us about why reward calibration key is to alignment and how to do RLHF for test-time scaling
Replying to @abeirami
[Thu Jul 17]
w/ @ananthbshankar & @jacobeisenstein, we present a reinforcement learning framework in view of test-time scaling. We show how to optimally calibrate & transform rewards to obtain optimal performance with a given test-time algorithm.
xitter.cf/SZiteng/status/1889005…
[Today 11 am poster E-2804 #ICML2025] Inference-time compute have been instrumental to recent development of LLMs. Can we align our model to better suit a given inference-time procedure? Come check our poster and discuss with @ananthbshankar, @abeirami, @jacobeisenstein, and myself.
Check out this thread for a short intro: xitter.cf/SZiteng/status/1889005…
Inference-time procedures (e.g. Best-of-N, CoT) have been instrumental to recent development of LLMs. The standard RLHF framework focuses only on improving the trained model. This creates a train/inference mismatch.
Can we align our model to better suit a given inference-time procedure?
We answer this affirmatively, check out the thread below.
Paper link: arxiv.org/abs/2412.19792.
Ziteng Sun retweeted
Inference-time procedures (e.g. Best-of-N, CoT) have been instrumental to recent development of LLMs. The standard RLHF framework focuses only on improving the trained model. This creates a train/inference mismatch.
Can we align our model to better suit a given inference-time procedure?
We answer this affirmatively, check out the thread below.
Ziteng Sun retweeted
Announcing the first workshop on Foundations of Post-Training (FoPT) at COLT 2025!
📝 Soliciting abstracts/posters exploring theoretical & practical aspects of post-training and RL with language models!
│
🗓️ Deadline: May 19, 2025
Ziteng Sun retweeted
Today at 10am I will present @SZiteng's paper
"block verification accelerates speculative decoding"
Replying to @abeirami
Friday 10am, I will present @SZiteng's paper on 𝐛𝐥𝐨𝐜𝐤 𝐯𝐞𝐫𝐢𝐟𝐢𝐜𝐚𝐭𝐢𝐨𝐧 𝐟𝐨𝐫 𝐬𝐩𝐞𝐜𝐮𝐥𝐚𝐭𝐢𝐯𝐞 𝐝𝐞𝐜𝐨𝐝𝐢𝐧𝐠 (w/ @th33rtha)
xitter.cf/SZiteng/status/1770845…
Ziteng Sun retweeted
Jointly announcing EAGLE-3 with SGLang: Setting a new record in LLM inference acceleration!
- 5x🚀than vanilla (on HF)
- 1.4x🚀than EAGLE-2 (on HF)
- A record of ~400 TPS on LLama 3.1 8B with a single H100 (on SGLang)
- 1.65x🚀in latency even for large bs=64 (on SGLang)
- A new scaling law: more training data, better speedup
- Apache 2.0
Paper: arxiv.org/abs/2503.01840
Code: github.com/SafeAILab/EAGLE
SGLang version: github.com/sgl-project/sglan…
⚒️Takeaway: Introducing training-time test, a novel draft model training technique: we replace feature prediction with direct token prediction and shift from top-layer-only features to multi-layer feature fusion. This approach unlocks a new scaling law previously undiscovered in EAGLE and EAGLE-2.
🙏Acknowledge: We would like to thank the SGLang team (@zhyncs42 @lm_zheng @ying11231 @JamesLiuID, @ispobaoke, and others @lmsysorg) for their merge and careful evaluation of EAGLE-3 on SGLang.
🤝Want to collaborate? We're a small academic group with limited GPU resources. If you're interested in supporting our next version of EAGLE or would like us to train a preliminary version tailored to a specific model, please get in touch!
Joint work with Yuhui Li, Fangyun Wei, and Chao Zhang
Ziteng Sun retweeted
⏰📢After years of working on long-context efficiency, I’ve started to doubt if it’s truly necessary (Many of you have probably noticed the decline of interest in long llms). Despite strong models like Gemini, short-context + retrieval often do the trick—faster, cheaper, and simpler. Are we overhyping super-long contexts?
Our recent work GSM-♾ changed my mind. A good long-context reasoning benchmark beyond retrieval/memory is the key to breaking this cycle. Such a relief 😃! Now, I truly believe stronger long-context models are essential for future intelligent agents.
I'm so excited about this work & all the great progress ahead 🚀~ Happy Valentine's Day 💗
🚀 RAG vs. Long-Context LLMs: The Real Battle ⚔️
🤯Turns out, simple-to-build RAG can match million-dollar long-context LLMs (LC LLMs) on most existing benchmarks.
🤡So, do we even need long-context models?
YES. Because today’s benchmarks are flawed:
⛳ Too Simple – Over-reliant on retrieval & QA.
⛳ Detectable Noise – RAG can filter out filler text easily.
⛳ Too Few – High-quality data requires huge human effort.
🔭 With LC LLMs hitting the ceiling, we need a benchmark that justifies their insane training costs.
🔥 Introducing 🐭🐷 GSM-Infinite – our synthetic long-context reasoning benchmark built to push LLMs to their real limits.
💎 Infinitely scalable in reasoning complexity & quantity
💎 Precision control over reasoning complexity
💎 Fully customizable RAG-proof context lengths 🚀
[1/n]
📄Paper: arxiv.org/abs/2502.05252
🖥️Code: github.com/Infini-AI-Lab/gsm…
🤗Huggingface datasets: huggingface.co/collections/I…
🏃Leaderboard: huggingface.co/spaces/Infini…
Inference-time procedures (e.g. Best-of-N, CoT) have been instrumental to recent development of LLMs. The standard RLHF framework focuses only on improving the trained model. This creates a train/inference mismatch.
Can we align our model to better suit a given inference-time procedure?
We answer this affirmatively, check out the thread below.
We also show that our proposed reward calibration method is a strong baseline for optimizing standard win rate on all considered datasets, with comparable or better performance than other SOTA methods, demonstrating the benefits of the reward calibration step.
Check out the paper for more details: arxiv.org/pdf/2412.19792.
Joint work w/ wonderful colleagues at Google: @ananthbshankar (co-lead), @JonathanBerant, @jacobeisenstein, Michael Collins, Adrian Hutter, Jong Lee, @nagpalchirag, @FlavienProst, @aradsinha, @th33rtha, and @abeirami.




























