Infini-AI-Lab (@InfiniAILab) header
Infini-AI-Lab (@InfiniAILab) profile picture
Pittsburgh, PA
Joined September 2024
Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)Photo by Infini-AI-Lab (@InfiniAILab)
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
2
16
5
87
9,971
A harness adapter in action: five Codex agents collaborating to prove a Lean theorem. The agents communicate through the adapter, sharing plans, messages, and verified lemmas. Agents split the proof into subproblems, prove lemmas in parallel, and reuse one anotherโ€™s resultsโ€”all without modifying Codex itself. 4/5
1
2
190
Make the training loop your own. Connect your task execution, feedback, and adapter update logic through Hermitโ€™s interfaces. Reuse its checks, versioning, and run records as you experiment with new evolution methods. Hermit is open source. Try it on your tasks, build an Evolver, and share what you want your agents to learn next. GitHub: github.com/Infini-AI-Lab/herโ€ฆ Website: infini-ai-lab.github.io/hermโ€ฆ 5/5
1
1
159
The gains extend beyond knowledge recall. Compared with continued dense training under the same token budget, STAPLE improves OLMo-2 1B by: โ€ข +4.27 points on HumanEval โ€ข +2.80 on MBPP โ€ข +2.27 on GSM8K โ€ข +1.65 on DROP STAPLE achieves a higher ten-task average than the static sparse alternatives such as STEM, Engram-style, as well as a serving-oriented MoE upcycling baseline with eight experts and top-1 routing. We use 100B tokens for warmup and another 100B for midtraining. The benefits extend to OLMo-2 7B despite shorter training budget (30B warmup + 70B midtraining): average math performance improves by 4.2% and code by 3.7% relative to dense continuation, with improvements on 9 of 10 tasks. MMLU gains are smaller, potentially reflecting the shorter warmup.
1
1
137
Billions of added parameters do not have to mean billions of extra parameters resident on GPU. STAPLE-12L keeps 19.73 GB of embedding tables on CPU, using a 1 GB GPU cache. Predictable access enables prefetching; sparse training updates touch only the rows used by the batch. The result: โ€ข Decoding throughput within 1.5% of dense โ€ข Training throughput within 3% of dense โ€ข About 2ร— the decoding throughput of our MoE baseline โ€ข 58% lower GPU energy per token than our MoE baseline, with only 1.2% overhead over dense
1
1
133
What does the new memory actually contribute? Removing the learned embeddings sharply degrades math and code performance, showing that the model has learned to rely on them. The memory is interpretable and editable, while remaining responsive to context - a promising foundation for expanding pretrained models with additional complementary knowledge.
1
124
๐Ÿง  How can we keep scaling LLMs with minimal changes in their serving energy? DeepSeekโ€™s Engram and our concurrent work STEM showed the promise of static sparse memory: large embedding tables with efficient access. But can mature pretrained models learn to use a newly added memory within a limited training budget, avoiding costly pretraining from scratch? ๐Ÿš€ Introducing STAPLE: a recipe to extend pretrained models' abilities with static sparse memory. Serving energy per token stays close to dense and >50% below MoE upcycling in high-throughput regime. Scalable capacity, interpretable memory, and compatibility with MoE upcycling - without any meaningful increase in serving cost. ๐Ÿงต Paper: arxiv.org/abs/2610.12599 Code: github.com/Infini-AI-Lab/STAโ€ฆ Website: infini-ai-lab.github.io/STAPโ€ฆ
3
18
5
51
5,537
Our central insight: under a limited training budget, new memory should complement pretrained knowledge. STEM replaces FFN up-projections with token-indexed embeddings. For upcycling, STAPLE retains those projections and adds the embeddings alongside them, under the same context-dependent gate. The recipe: โ€ข Zero-initialize the embeddings to preserve the starting modelโ€™s function. โ€ข Warm up with the retained up-projections frozen. โ€ข Then jointly mid-train the model and memory. Retaining the pretrained pathway makes a substantial difference. Warmup further improves knowledge and commonsense performance. The architecture and training schedule both matter.
1
1
216
Why static sparsity for expanding pretrained models? MoE upcycling can add capacity without restarting pretraining. But cloned experts can converge to homogeneous experts, might need additional tricks and more training budget to differentiate, while larger expert pools increase GPU-memory demands and can slow serving. Static memory offers input-specific lookups with no expert homogeneity issues. Additionally, they can keep large tables on CPU and efficiently retrieve only the needed rows. The challenge is making that memory useful quickly. Token frequencies are highly uneven, leaving rare rows with few updates. Replacing pretrained computation also discards useful features that must be relearned. How do we add capacity while preserving what the model already knows?
1
1
399
Infini-AI-Lab retweeted
Serving data is an underestimated gold mine for agent self-improvement. But how well can agents actually learn from their serving experience? In our recent work, ServeLearnBench, we take a closer look at this question. We find that current agents still struggle to learn from experience, and effective exploration plays a surprisingly important role. Check out more results in our paper and repo! ๐Ÿ“„ Paper: arxiv.org/abs/2610.07792 ๐Ÿ’ป Code & Dataset: github.com/Infini-AI-Lab/Serโ€ฆ ๐ŸŒ Website: infini-ai-lab.github.io/Servโ€ฆ
How well can agents continually self-improve from serving experience? Weโ€™re excited to introduce our new benchmark, ๐’๐ž๐ซ๐ฏ๐ž๐‹๐ž๐š๐ซ๐ง๐๐ž๐ง๐œ๐ก: Real-world deployment is not a one-time test of capability. The knowledge agents need is often hidden and can change over time. ServeLearnBench tests whether agents can infer, apply, and revise this knowledge from interaction. Across 5 learning harnesses ร— 6 models, we find: 1. ๐‹๐š๐ซ๐ ๐ž ๐ฅ๐ž๐š๐ซ๐ง๐ข๐ง๐  ๐ ๐š๐ฉ๐ฌ ๐ซ๐ž๐ฆ๐š๐ข๐ง. Agents can solve tasks when the hidden policy is given, but struggle to discover it from experience. 2. ๐๐จ ๐Ÿ๐ซ๐ž๐ž ๐ฅ๐ฎ๐ง๐œ๐ก ๐Ÿ๐จ๐ซ ๐š๐๐š๐ฉ๐ญ๐š๐ญ๐ข๐จ๐ง. Continual learning can be costlyโ€”and can even degrade already-correct behavior. 3. ๐„๐ฑ๐ฉ๐ฅ๐จ๐ซ๐š๐ญ๐ข๐จ๐ง ๐ข๐ฌ ๐š ๐ค๐ž๐ฒ ๐›๐จ๐ญ๐ญ๐ฅ๐ž๐ง๐ž๐œ๐ค. Better exploration strongly correlates with better adaptation.
4
12
653
Infini-AI-Lab retweeted
The next frontier would be agents that get smarter as you use them. ๐Ÿš€ Our new ServeLearnBench puts continual self-improvement to the testโ€”and reveals whatโ€™s holding it back.
How well can agents continually self-improve from serving experience? Weโ€™re excited to introduce our new benchmark, ๐’๐ž๐ซ๐ฏ๐ž๐‹๐ž๐š๐ซ๐ง๐๐ž๐ง๐œ๐ก: Real-world deployment is not a one-time test of capability. The knowledge agents need is often hidden and can change over time. ServeLearnBench tests whether agents can infer, apply, and revise this knowledge from interaction. Across 5 learning harnesses ร— 6 models, we find: 1. ๐‹๐š๐ซ๐ ๐ž ๐ฅ๐ž๐š๐ซ๐ง๐ข๐ง๐  ๐ ๐š๐ฉ๐ฌ ๐ซ๐ž๐ฆ๐š๐ข๐ง. Agents can solve tasks when the hidden policy is given, but struggle to discover it from experience. 2. ๐๐จ ๐Ÿ๐ซ๐ž๐ž ๐ฅ๐ฎ๐ง๐œ๐ก ๐Ÿ๐จ๐ซ ๐š๐๐š๐ฉ๐ญ๐š๐ญ๐ข๐จ๐ง. Continual learning can be costlyโ€”and can even degrade already-correct behavior. 3. ๐„๐ฑ๐ฉ๐ฅ๐จ๐ซ๐š๐ญ๐ข๐จ๐ง ๐ข๐ฌ ๐š ๐ค๐ž๐ฒ ๐›๐จ๐ญ๐ญ๐ฅ๐ž๐ง๐ž๐œ๐ค. Better exploration strongly correlates with better adaptation.
1
17
2
117
9,269
How well can agents continually self-improve from serving experience? Weโ€™re excited to introduce our new benchmark, ๐’๐ž๐ซ๐ฏ๐ž๐‹๐ž๐š๐ซ๐ง๐๐ž๐ง๐œ๐ก: Real-world deployment is not a one-time test of capability. The knowledge agents need is often hidden and can change over time. ServeLearnBench tests whether agents can infer, apply, and revise this knowledge from interaction. Across 5 learning harnesses ร— 6 models, we find: 1. ๐‹๐š๐ซ๐ ๐ž ๐ฅ๐ž๐š๐ซ๐ง๐ข๐ง๐  ๐ ๐š๐ฉ๐ฌ ๐ซ๐ž๐ฆ๐š๐ข๐ง. Agents can solve tasks when the hidden policy is given, but struggle to discover it from experience. 2. ๐๐จ ๐Ÿ๐ซ๐ž๐ž ๐ฅ๐ฎ๐ง๐œ๐ก ๐Ÿ๐จ๐ซ ๐š๐๐š๐ฉ๐ญ๐š๐ญ๐ข๐จ๐ง. Continual learning can be costlyโ€”and can even degrade already-correct behavior. 3. ๐„๐ฑ๐ฉ๐ฅ๐จ๐ซ๐š๐ญ๐ข๐จ๐ง ๐ข๐ฌ ๐š ๐ค๐ž๐ฒ ๐›๐จ๐ญ๐ญ๐ฅ๐ž๐ง๐ž๐œ๐ค. Better exploration strongly correlates with better adaptation.
1
8
4
42
11,019
Yet a substantial learning gap remains. Even when models have the capability to solve the task, they struggle to learn the required behavior from serving experience. On Retail L3, the best learner reaches only 59.6%. And adaptation isnโ€™t free: learning harnesses cost 4โ€“101ร— as much as Oracle and can even hurt already-correct behavior. ๐Ÿ’ธ ๐Ÿงต(4/5)
1
2
331
Explore ServeLearnBench: ๐–๐ž๐›๐ฌ๐ข๐ญ๐ž: infini-ai-lab.github.io/Servโ€ฆ ๐‚๐จ๐๐ž&๐ƒ๐š๐ญ๐š๐ฌ๐ž๐ญ: github.com/Infini-AI-Lab/Serโ€ฆ ๐๐š๐ฉ๐ž๐ซ: arxiv.org/abs/2610.07792 ๐Ÿงต(5/5)
2
269
We test the agent on 5 real-time apps, each deployed from a reference alone: face-to-face conversational avatar, a multimodal LLM (Qwen3-Omni) in text-prompt/speech-response mode, a video background editor, a video world model, and a video narrator. Using Claude Code with Opus 4.8, FlashRT produces efficient real-time deployments for each one, reaching up to ~70ร— reduction in response latency. Its throughput gains are largest on AMD MI355X (up to 3.6ร— frame rate, versus 2.8ร— on B200), showing that FlashRT enables more scalability on platforms there's less mature expert optimization. ๐Ÿงต5/5
1
3
209
Joint work with @wonderingkrish, @chenzhuoming911, Yanyuan Qin, Zhenyu Gu, Atri Rudra, and @BeidiChen. We are grateful for the collaboration with @AMD!
3
157