
Member of Technical Staff, AGI Autonomy @amazon. Building artificial general intelligence. Opinions are my own.
San Francisco, CA
Joined April 2022
- Tweets346
- Following1.2K
- Followers452
- Likes1.3K
Wai Tong retweeted
What if your agent could switch harnesses mid-run?
Imagine starting with Claude Code, then switching to Pi or Codex. Would it approach the task differently? Catch mistakes it previously missed? Could different harnesses complement each other? 🧵
Wai Tong retweeted
check out our recent work on cascade harnesses! i think more so than designing any one harness, it's important to understand what structures will maximize performance at low cost. this was a great collab!
Wai Tong retweeted
As a researcher, I cannot understate how much easier this determinism makes my life. No more mystery "maybe it's just distributed computing?? 🤷♀️" results from my dataloader
Zephon’s TorchTitan integration preserves identical global batches when training resumes on a different GPU count.
In our latest blog’s 1B-parameter model test, training resumed from a 16-GPU checkpoint on 8, 32, and 64 GPUs, and the SHA-256 hash of every global batch matched the original run.
Zephon tokenizes, packs, shuffles, and mixes data inside the training job, avoiding a preprocessed copy of the corpus.
The data, tokenizer, seed, mixture, context length, and global batch size must stay unchanged when resuming. The lane count must divide evenly across data-parallel replicas, and each optimizer step’s batch must split evenly across lanes. Training losses can still differ because floating-point reduction order depends on GPU count.
✍️ @datologyai @josh_wills @MaxiBoether, Ties Robroek (@datologyai)
Read more: pytorch.org/blog/bringing-el…
Wai Tong retweeted
Spent the past three weeks on this idea.
A sampler on a different precision or accelerator is just off-policy, like stale samples in fully async RL, and off-policy correction doesn't care why the sampler differs from the trainer. Meanwhile, inference keeps diversifying across precisions(MXFP8, MXFP4, NVFP4) and accelerators outside RL. So why not use all of that compute for RL rollout?
The sampler and the trainer are two systems built for different purposes. Hope this opens up new possibilities for RL algorithms and systems, and leads to cleaner system designs. Enjoy the blog, and stay tuned for the next one on our infra :)
Training large models with RL typically runs on a fixed homogeneous cluster of the latest accelerators for numerical stability and faster throughput. This throws away inference optimization work done for other vendors and generations, limits the capacity you can utilize, and is unnecessarily expensive.
So we trained a model on agentic coding tasks using fully asynchronous RL and mixed-precision, mixed-accelerator samplers. With off-policy correction, training stayed stable through convergence, on par with homogeneous samplers, at a lower cost.
Wai Tong retweeted
Amazing work from @modal and @nanjiangwill !
A refreshing glimpse into the future of large-scale RL!
Training large models with RL typically runs on a fixed homogeneous cluster of the latest accelerators for numerical stability and faster throughput. This throws away inference optimization work done for other vendors and generations, limits the capacity you can utilize, and is unnecessarily expensive.
So we trained a model on agentic coding tasks using fully asynchronous RL and mixed-precision, mixed-accelerator samplers. With off-policy correction, training stayed stable through convergence, on par with homogeneous samplers, at a lower cost.
Training large models with RL typically runs on a fixed homogeneous cluster of the latest accelerators for numerical stability and faster throughput. This throws away inference optimization work done for other vendors and generations, limits the capacity you can utilize, and is unnecessarily expensive.
So we trained a model on agentic coding tasks using fully asynchronous RL and mixed-precision, mixed-accelerator samplers. With off-policy correction, training stayed stable through convergence, on par with homogeneous samplers, at a lower cost.
Wai Tong retweeted
This result is super impressive ESPECIALLY considering the "baseline" here is crazy sophisticated
Replying to @datologyai
The gains persisted after post-training: our 30B-A3B model scored 46.8% vs. 37.7% for the baseline, averaged across benchmarks.
A curated 12B-A1.4B model also beat the 30B-A3B baseline despite using roughly one-fifth of the pretraining compute.
Wai Tong retweeted
Today, we’re excited to launch the Datology Curation Studio, customizable frontier data curation for every team training its own models.
Data quality is the ultimate compute multiplier, and Curation Studio makes it easy for everyone.
Wai Tong retweeted
Can confirm this works really well - amazing work @MaxiBoether and team!
At @datologyai we run thousands of data experiments. Two things keep biting us:
- Same run on 16 vs 8 GPUs → different evals
- Every pipeline change → re-materialize the dataset
Introducing Zephon, the data loader we built to process data on the fly, deterministically. 🧵
Wai Tong retweeted
My lab today at COLM: @GeorgeMorgulis presenting Subliminal Steering, 11:00-1:00 in Franciscan A #78
If you’re studying subliminal learning in LMs, you should be studying George’s variant, which is induced by steering. It just works way more consistently.
Proud to have contributed to this RL repo via the open source
Today we're launching Miles v0.1, an open-source RL framework for LLMs and multimodal models.
RL training is easy to start and hard to debug. Miles helps you ensure your run is correct, use hardware efficiently, and keep RL running at scale.
Over the past 9 months, 72 contributors have landed 1,326 commits, 85 GPU E2E CI tests, battle-testing Miles on frontier open models like Kimi K3, DeepSeek V4, Qwen 3.8, GLM 5.2, Inkling, MiniMax H3, etc.
Miles powers frontier-model development and production RL workloads at @humansand, @periodiclabs, @modal, @DecagonAI, @Eigent_AI, @nebiusai, @IBM and more, on both @NVIDIAAI and @AIatAMD hardware.
Here is what we built, and why teams picked Miles🧵
In collaboration with @togethercompute, IBM is scaling open source AI.
Together AI will leverage @NVIDIA HGX B300 systems and NVIDIA Spectrum-X networking on IBM Cloud to deliver open source model inference, helping enterprises run AI workloads faster and more efficiently in production: ibm.co/6017ERyy7
Wai Tong retweeted
Turnstile is an open-source Rust proxy that records token-native rollout data for agentic RL. Harness stays unchanged. Trainer gets exact token IDs, log probs, and loss masks. amazon.science/blog/capturin…
Wai Tong retweeted
super excited about Sailboxes - the best home for ambitious AI agents! @charley3c and @NirvikBaruah did an incredible job here. You know it's real when minecraft works ⛏️
Today we're launching Sailboxes: the first cloud environment purpose-built for long-horizon AI agents. Sailboxes are full machines with persistent state that auto-sleep, built for agents that live forever, priced so you only pay for what they actually use. From $0.015 per active vCPU-hour, over 70% cheaper than other providers.
Under the hood: novel architecture from @NirvikBaruah and @charley3c that live-migrates VMs based on actual resource usage. To show how robust it is, we ran a Minecraft server in a Sailbox and forced it to migrate every 2 minutes--players never noticed.
Wai Tong retweeted
Today we're launching Sailboxes: the first cloud environment purpose-built for long-horizon AI agents. Sailboxes are full machines with persistent state that auto-sleep, built for agents that live forever, priced so you only pay for what they actually use. From $0.015 per active vCPU-hour, over 70% cheaper than other providers.
Under the hood: novel architecture from @NirvikBaruah and @charley3c that live-migrates VMs based on actual resource usage. To show how robust it is, we ran a Minecraft server in a Sailbox and forced it to migrate every 2 minutes--players never noticed.
Wai Tong retweeted
I’m excited to share something I’ve been working on: Turnstile, a Rust proxy for token-level rollout capture in agentic RL.
It sits between a Chat Completions harness and model backend, recording everything trainers need.
github.com/amazon-agi-labs/t…
It's been great working with @_frrad at @AmazonScience and I have no doubt that turnstile will leave a great impact to the agentic AI ecosystem!
Read more here: amazon.science/blog/capturin…
I’m excited to share something I’ve been working on: Turnstile, a Rust proxy for token-level rollout capture in agentic RL.
It sits between a Chat Completions harness and model backend, recording everything trainers need.
github.com/amazon-agi-labs/t…

























