Zhihao Zhang (@Jackfram2) profile picture
Joined May 2019
Zhihao Zhang retweeted
2,100+ tokens/sec per user on DeepSeek V4.1 Flash, at full model quality, on 8× AMD Instinct MI355X. 🚀 Congrats to @lithos_ai on pushing the frontier of inference with persistent megakernels on AMD. Open hardware + instruction-level control = room to innovate.
2,100+ tokens per second per user on DeepSeek V4.1 Flash, at full model quality, on eight AMD Instinct MI355X GPUs. In our benchmarks, the Lithos engine delivers: → Up to 3.4× faster inference for DeepSeek V4.1 Flash → Up to 2.5× faster inference for Kimi K3 Compared with tested open-source engines on the same eight GPUs. The key challenge is reducing the coordination tax: time spent waiting between kernels and across GPUs. Our approach combines persistent megakernels, overlapping execution, and communication inside computation. Our technical deep dive covers: - Decode speed, cost per token, and latency up to 1M context. - How Lithos engine reduces launch overhead and overlap computation with communication across GPUs. - How MI355X’s register capacity and direct instruction-level control enable hardware-specific optimizations. - What our roofline analysis reveals about opportunities for further speedups. Read the full analysis: lithosai.com/blog/ultra-fast…
17
41
14
491
47,101
Zhihao Zhang retweeted
2,100+ tokens per second per user on DeepSeek V4.1 Flash, at full model quality, on eight AMD Instinct MI355X GPUs. In our benchmarks, the Lithos engine delivers: → Up to 3.4× faster inference for DeepSeek V4.1 Flash → Up to 2.5× faster inference for Kimi K3 Compared with tested open-source engines on the same eight GPUs. The key challenge is reducing the coordination tax: time spent waiting between kernels and across GPUs. Our approach combines persistent megakernels, overlapping execution, and communication inside computation. Our technical deep dive covers: - Decode speed, cost per token, and latency up to 1M context. - How Lithos engine reduces launch overhead and overlap computation with communication across GPUs. - How MI355X’s register capacity and direct instruction-level control enable hardware-specific optimizations. - What our roofline analysis reveals about opportunities for further speedups. Read the full analysis: lithosai.com/blog/ultra-fast…
22
8
10
182
32,713
Zhihao Zhang retweeted
We are excited to introduce 𝐇𝐞𝐫𝐦𝐢𝐭 🐚: an open-source framework for training, evaluating, and deploying harness adapters. Harness evolution is a key building block for RSI and continual learning, enabling agents to continually improve how they work from experience. Train a reusable external layer. Use existing evolution methods or build your own through one standard framework. GitHub: github.com/Infini-AI-Lab/her… 1/5 #HarnessEvolution #ContinualLearning #RSI #Agent
1
15
5
91
10,243
Zhihao Zhang retweeted
Introducing Kimi K3.1 Pro: our most capable 3T-class open-weight model. 🔹 1M-token context, 2.8T total / 104B active params (MoE), Multimodal. 🔹 Full-project coding: games in Unreal, Unity & Godot. 🔹 Excels at parametric 3D modeling in Blender & CAD. 🔹 More reliable at finance work: Accurate modeling with hallucination down by 65%. 🔹 Matching the strongest proprietary models across coding, agentic and knowledge benchmarks. Kimi K3.1 Pro and K3.1 Pro Highspeed (2–3× faster) are now live on Kimi.ai, Kimi Work, Kimi Code, and the Kimi API. Open weights by Nov 4, 2026. 🔗 API: platform.kimi.ai 🔗 Tech blog: kimi.ai/blog/kimi-k3-1-pro
1,279
3,231
2,371
30,250
4,160,409
Zhihao Zhang retweeted
Exciting news: DeepSeek-V4.1-Flash-20261017 (Max) by @deepseek_ai is now the #1 open model in Code Arena: WebDev, surpassing Kimi K3 (Max) at 10x lower cost! Highlights: - #1 open model, ahead of Kimi K3 (Max) - #12 overall with 1670 pts, ahead of Claude Opus 5 (High) and within ~10 pts of GPT-6 Sol (Max) - Shifts the Pareto frontier at $0.3/$1.2 per million input/output tokens - +52 pts vs. the previous DeepSeek-V4.1-Flash-20260910 Congrats to the @deepseek_ai team on another strong contribution to the open source ecosystem!
🤖 Made with AI
🚀 DeepSeek-V4.1-Flash-1017 is here! Stronger benchmark performance, faster task completion, and shorter outputs than the previous version—a Pareto improvement in performance and cost. Now live on our website, app, and API. 🛠️ Optimized for @DeepSeekHarness, Tencent CodeBuddy, WorkBuddy, and more. Upgrade to DeepSeek Harness v0.2.1-rc.1 or later for the best experience with the new model. 1/4
111
191
70
2,473
284,069
Zhihao Zhang retweeted
DeepSeek-V4.1-Flash-1017 is now live on LithosAI at up to 1,500 tok/s 🚀 Great model upgrade from @deepseek_ai
🚀 DeepSeek-V4.1-Flash-1017 is here! Stronger benchmark performance, faster task completion, and shorter outputs than the previous version—a Pareto improvement in performance and cost. Now live on our website, app, and API. 🛠️ Optimized for @DeepSeekHarness, Tencent CodeBuddy, WorkBuddy, and more. Upgrade to DeepSeek Harness v0.2.1-rc.1 or later for the best experience with the new model. 1/4
5
4
2
27
1,751
Zhihao Zhang retweeted
🚀 What if formal verification could make GPU kernels both safer and faster? Introducing SentryWard, an open-source formal verifier for GPU kernels. 👁⚡
15
33
7
170
14,654
Zhihao Zhang retweeted
Optimizing local inference for ultra-low latency is both easier and harder than optimizing for a cloud service. The unavoidable hardware variety, memory constraints, and thermal throttling automatically make each optimization more complicated. But knowing which challenges to tackle first is easier because you’re dealing with users who: 1) know exactly what they’re looking for from inference performance for their setup and 2) are predisposed to build in the open and share their findings Here’s how we know: @JiaZhihao announced that we were open-sourcing lithos-metal, our local inference engine for Apple silicon, 5 days ago. In that time, we’ve made a ton of progress because everyone who let us know they tried lithos-metal (or even just explored the code or blog) gave us the platonic ideal of technical feedback: immediate, specific, actionable. Thanks to all of you who shared your questions and suggestions, including insights into your actual use cases and testing approach. Our day job is still primarily building ultra low-latency agent infrastructure on standard GPUs, but it’s important to us to keep lithos-metal going and build the community. Internally, we talk a lot about taking a co-design approach to AI efficiency. That means simultaneously thinking about how the hardware and software layers impact each other. Only focusing on one area for improvement leaves performance gains on the table. That logic applies to where your inference engine lives, too. Whether it’s in the cloud, on-prem, or on your laptop, we’re excited to keep working to solve agent latency for our users.
🤖 Made with AI
2
6
16
1,368
Zhihao Zhang retweeted
We’re open-sourcing lithos-metal 🚀 Megakernels + DSpark speculative decoding run Qwen3.8-27B at 200+ tokens/s/user peak on one @Apple M5 Max. Ultra-fast inference on your laptop. Try it with any coding agent in one command. Code: github.com/lithos-ai/lithos-… Tech blog: lithosai.com/blog/lithos-met…
51
76
20
578
423,372
Zhihao Zhang retweeted
We’ve all seen the cool “Astra for robotics” demos. But behind the scenes? Slow rollouts, hand-engineered APIs, and/or simulated behaviors that struggle on real robots. How do we make coding agent actually work in the physical world? 🦾 Introducing SimEX: Simulation-Integrated Robotics AutoResearch 🧵 arxiv.org/abs/2609.38982
11
60
8
295
32,518
Zhihao Zhang retweeted
Three models, three #1 spots on @ArtificialAnlys: Kimi K3, DeepSeek V4.1 Flash, and GLM-5.3-Flash. 🚀 Together with LithosBox, we’re accelerating the full agent loop with the fastest inference and sandbox execution. Try @lithos_ai on your workloads!
Another day, another LithosAI launch, and another #1 spot on @ArtificialAnlys benchmarks for inference output speed and end-to-end latency. That’s 3 for 3: LithosAI has the fastest inference for Kimi K3, DeepSeek V4.1-Flash, and now GLM 5.3 Flash. Combine that with LithosBox, the ultra-fast agent sandboxes we just launched, and our users can significantly speed up the agent loop. Get started with our public API.
5
21
1,933
Zhihao Zhang retweeted
We will be teaching Agentic GPU Programming for MLSys as part of the CMU ML Systems course this upcoming spring at @SCSatCMU, touching on knowledge curation, compiler analysis, and feedback loops. We also curated an online book on the topic, check it out mlc.ai/agentic-gpu-programmi…
24
165
8
1,070
81,556
Zhihao Zhang retweeted
LithosAI is pushing the Pareto frontier of agent inference speed, latency, and cost, and our latest results on Artificial Analysis proves it. Unlike most providers, offering more tiers improves our utilization. Read the full story:
2
7
3
34
478,950
Zhihao Zhang retweeted
🍰 CAKE paper's out, the design bet: the compiler isn't a fixed black box the agent calls — it's part of the harness, and it's under evolution too. CAKE didn't inherit existing abstraction layer. no tile/layout abstractions: the vocabulary was distilled by agents from a corpus of production kernels. every pattern the agent couldn't express pushed new primitives into the IR, and the analyses to keep them checkable. every barrier/layout bug that kept coming back became a verifier rule. none of this can be designed up front. the IR has to co-evolve with the kernels, and the workload tells you what's missing, the corpus tells you if the fix broke anything. the best language for an agent is the one that tells you what's illegal, what's slow, and which decision might made it faster. arxiv.org/abs/2608.12629
13
82
23
323
150,148
Zhihao Zhang retweeted
For my first post, I’m sharing a letter @NVIDIA signed on why open models matter. AI will transform every industry, power every company, and be built by every country. Open models strengthen safety and cybersecurity, accelerate innovation and diffusion, and enable sovereignty. The world needs both frontier closed models and frontier open models. images.nvidia.com/pdf/Open-W…
15,950
29,203
10,949
171,194
66,532,281
Zhihao Zhang retweeted
One year since we released MPK, and the momentum around megakernels keeps growing. 🎉A few exciting updates: If you are at #OSDI2026, @chengxinhao1 will present MPK today. Come learn how we compile LLMs into a single megakernel. We've rebuilt MPK's runtime from the ground up for Blackwell. Faster, leaner megakernels are coming, so stay tuned for the release. The community is extending megakernels to more hardware backends. Check out this port to AMD MI350X arxiv.org/pdf/2604.15379v1 Grateful to our 50+ contributors building with us. The megakernel era is just getting started. 🚀
One of the best ways to reduce LLM latency is by fusing all computation and communication into a single GPU megakernel. But writing megakernels by hand is extremely hard. 🚀Introducing Mirage Persistent Kernel (MPK), a compiler that automatically transforms LLMs into optimized megakernel, reducing latency by 1.2-6.7x. 🔧Tool: github.com/mirage-project/mi… 📝Blog: zhihaojia.medium.com/compili…
2
1
35
4,626
Zhihao Zhang retweeted
Today we're launching Lithos’ agentic inference engine starting with Kimi K2.7 Code, a frontier open-source 1T parameter coding model. On a single standard 8xB200 GPU node, performance peaks above 1,000 tok/s per user on coding workloads at model-native precision, with no approximation and full model quality. Raw speed is the agent differentiator now. Check it out at lithosai.com
6
9
6
95
127,179
Zhihao Zhang retweeted
We taught a brand-new mini-series this year at @SCSatCMU on Modern GPU Programming for ML Systems, as part of the ML Systems course, touching on fun questions like what data layout swizzling is, how to use 3D TMA, and state-of-the-art Blackwell programming. We released a curated online book based on the materials: mlc.ai/modern-gpu-programmin… check it out
24
289
6
1,825
205,524
RT @ruihanglai: Two moments every ML researcher knows. You get onto a new cluster, and week one goes to fitting the framework to your setup…
9
14
Zhihao Zhang retweeted
Excited to share XGrammar-2 has been accepted by #ACMCAIS 2026 @CAISconf! 🎉 ⚡️ Up to 80x speedup 🛡️ Strict tool-calling correctness 🚀 Trusted by xAI in production systems Join our presentation today at 4 PM at Bayshore Ballroom to see how we power critical agent workloads at scale. 🔗Blog: blog.mlc.ai/2026/05/04/xgram… 🔗GitHub: github.com/mlc-ai/xgrammar
Introducing XGrammar-2: structured generation for complex agent harnesses. Strict tool-calling formats. Built-in DeepSeek-V4 and Qwen-3.6 support. Up to 80x speedup over XGrammar. Ready-to-use integrations with vLLM, SGLang, TensorRT-LLM, and more! ⚡ From Claude Code to OpenClaw, agents are defining more complex harnesses. XGrammar-2 ensures LLMs always interact with them in the right way. Built in collaboration with DeepSeek, Databricks, and leading frontier AI labs to bring XGrammar-2 into latest models and products. 🧩 Structural Tag: one unified abstraction to describe any format your agent needs 🚀 Scales to 500+ strictly typed tools for complex agent harnesses 🌐 Native APIs in Python, C++, Rust, and JS, running everywhere from cloud to edge 🛠️ Integrated with vLLM, SGLang, TensorRT-LLM, and more Excited to see what agent builders create with it! Blog: blog.mlc.ai/2026/05/04/xgram… GitHub: github.com/mlc-ai/xgrammar
1
12
40
4,442