Founder & CEO of an AI Research & Engineering Labs | Weekly @twimlai GenAI meetup | @Cohere_Labs Community Lead & Researcher | @AILucknow | Prev @TCS @Infobeans
๐ฅ TinyAya v0.3 follow-up: the run early-stopped โ so we launched the anneal leg it skipped. It's descending right now, and it's already beating the plateau.
For everyone who asked for the nitty-gritty after the last post โ here's what's actually happening on the TPU pod as I write this. ๐งต๐
Why an anneal leg exists at all. The 110,463-step run trained on a WSD (warmupโstableโdecay) schedule: 1,100 warmup steps, then a long plateau at lr 1.716e-4, with an 11,000-step linear decay to zero scheduled for the end. But we also ran stop-anytime early stopping โ 10 validation cycles (2,500 steps) without improvement ends the run. At step 65,250 of 110,463, the plateau flattened out (best val composite 2.9048 @ step 62,750) and the early stop firedโฆ which meant the scheduled decay never ran. Our round-3 ablation had shown the decay phase is where WSD harvests its loss drop โ it beat the plateau on every metric at probe scale. So the model was parked one phase short of done.
โ๏ธ The anneal leg, mechanically. We resumed the exact step-65,250 checkpoint โ weights, full Adam optimizer state, and RNG restored from GCS โ and extended max_steps to 76,250. With the WSD scheduler's total set to start+11,000, the resume lands directly inside the decay: no warmup replay, no discontinuity. Verified from the first logged points: lr = 1.7121e-4 at step 65,275, stepping down by exactly 3.9e-7 per 25 steps โ the predicted linear slope of 1.716e-4 รท 11,000. Every other hyperparameter is byte-identical to the plateau leg (global batch 32 = 2 rows/chip ร 16 chips, grad-clip 1.0, same seed, same 3,200-sample val gate every 250 steps), and the data order continues deterministically mid-epoch-2. Early stopping is OFF for the leg โ it's a fixed-length descent by design.
๐ Is it working? Yes โ measurably, within the first 750 steps.
โข plateau best: 2.9048 (step 62,750)
โข step 66,000: 2.8922
โข step 67,250: 2.8827 โ text token accuracy 95.2%, text loss 0.458, semantic-codebook accuracy 40.6%
Every validation cycle since the descent began has set a new best. ~9,000 steps (~6 h) still to go as the lr grinds toward zero.
๐ข Scale, in tokens: each step processes 32 sequences ร 300 frames ร 9 tokens/frame (1 text + 8 audio codebooks) โ 86k supervised tokens โ the anneal alone covers ~950M tokens, bringing the full run to ~6.6B by step 76,250. Hardware: the same multi-host TPU v6e-16 (4 hosts ร 4 chips, SPMD data-parallel), ~1.40 s/step, zero spot preemptions across the entire 2.2-day plateau leg.
๐งฏ One honest ops note for the monitoring nerds: W&B briefly showed the run as "crashed" mid-anneal โ it was a heartbeat lapse during the boot's one-time XLA compile of the checkpoint-save graph, not a failure. The process was at 200% CPU the whole time and recovered on its own. Multi-day TPU runs are 50% training and 50% learning which alarms are real.
๐ฆ Every checkpoint is kept (log-spaced early + every 1,000 steps) โ the full trajectory ships as a Pythia-style suite for anyone who wants to study how a speech-to-speech model learns, including how the anneal reshapes it.
๐ Watch the descent live: https://t.co/i0LMP0d1WX
๐ง The full engineering trail โ plan, hyperparameter rationale, resume drills (PR #10): https://t.co/PdbvUMGVYy
๐ Project blog: https://t.co/pgnC9Az03A
Release evals (ASR-chrF++, MOS, BLASER) and the Hugging Face suite land after the descent finishes. The next post has final numbers. ๐ฏ
TPUs via the TRC @Google grant and project in the @Cohere_Labs Expedition Tiny Aya 2026
#SpeechAI #MachineTranslation #TPU #LearningRateSchedules #OpenSource #MLOps
๐ฃ๏ธ๐ TinyAya v0.3 โ an open TurkishโHindi speech-to-speech translation model โ and the long-horizon training run is past its halfway mark.
๐ง v0.3 translates spoken Turkish to spoken Hindi (and back) directly in the audio-token domain: a Cohere2 LLM backbone (LoRA-fine-tuned) fused with Moshi's frozen depth decoder, generating Mimi codec tokens frame by frame โ text and 8 audio codebooks per 80 ms frame.
๐ Audio samples included as well.
โก The current run: 110,463 steps on a multi-host TPU v6e-16 pod (spot capacity via Google's TPU Research Cloud), now past 50% with roughly a day and a half to go โฑ๏ธ
๐ฆ Every checkpoint is kept โ log-spaced early saves plus every 1,000 steps โ so the full training trajectory ships as a Pythia-style suite for anyone who wants to study how a speech-to-speech model learns, not just the final weights.
๐ก๏ธ Making a multi-day spot-TPU run survivable was half the work, and that engineering trail is public: preemption self-healing with automatic resume, dataset preflight gates so a wrong corpus can never silently train, multi-host rendezvous, a coarse-to-fine codebook curriculum against deep-codebook collapse, and an 8-stage release-evals harness (ASR-chrF++, BLASER-2.0, MOS deltas, latency) with frozen, digest-verified eval sets.
๐ Live training run: https://t.co/i0LMP0d1WX
๐ง Full implementation & hardening story (PR #10): https://t.co/PdbvUMGVYy
๐ท๏ธ v0.3 code freeze: https://t.co/PHeyHaLSVk
๐ค Model checkpoints and eval numbers land on Hugging Face once the run completes and release evals pass. More soon.
#SpeechAI #MachineTranslation #TPU #OpenSource #SpeechToSpeech
Cohere Labs ๐คTwitch
When we started planning ML Summer School, we never imagined 7,000+ people from around the world would want to learn alongside us ๐คฏ
Because the response has been far bigger than we planned for, we're moving all sessions to a Twitch livestream so we can host all of you!
You'll still be able to earn a certificate of completion, so nothing changes there ๐
If you haven't registered yet, there's still time! Please register through Zoom so you'll receive all the session reminders, have the full schedule added to your calendar, and stay up to date throughout the week: https://t.co/kl8pzgG7SV
For over two years, the largest open code dataset was The Stack v2โฆ until today.
๐ฅ The Stack v3 is out: the largest open code dataset ever released: 114 TB, 770 languages, 224M repositories, ~5T tokens of deduplicated and filtered source code. Fully open, no restrictively licensed code included.
The upgrade:
- v2 (2024): 68 TB raw -> 2 TB / ~550B tokens, 618 languages filtered
- v3 (2026): 114 TB raw -> 15.9 TB / ~5T tokens, 713 languages filtered
C++ x15, TypeScript x7.5, Rust x7, Python x4.8. Even the COBOL corner of GitHub got a bigger slice.
Part of that is two fresh years of open source. Part of it is a bug we found in v2's deduplication - story below. ๐
Things that make v3 different:
1. Contents inline. The #1 complaint about v2 was "cool dataset, where's the actual code?" v2 shipped file IDs from the Software Heritage graph, and fetching contents was a DIY treasure hunt. v3 is self-contained: sources embedded directly, one row = one repository. Download finishes -> you start training.
2. Fresh crawl. v2 was a 2023 snapshot of even older crawls. v3 is a direct re-crawl of GitHub at the latest commit, completed by August 2025: 224M repositories, 44B files. Forks were only included if they had 5+ stars.
Two ways in:
๐ฅ stack-v3-train - near-deduplicated, quality-filtered, PII-redacted, contents inline. Point load_dataset at it and go.
https://t.co/px2LW3v6VB
๐ stack-v3-full - the entire 114 TB corpus as an HF Storage Bucket: every duplicate kept with cluster IDs, stubs for excluded files. Roll your own dedup, filters, and mixes.
https://t.co/SQzP5gOmzl
The record isn't really ours to claim - it's millions of everyone's repositories, neatly stacked. Thanks to every developer who keeps their code public, the BigCode community, and friends who built the infrastructure that made this possible!
Understanding Speculative Decoding
Deep dive into speculative decoding feat. Evan Carlson
From the ICML 2023 Oral @ https://t.co/GPlr4lWyG1
https://t.co/zHpwhnKJKO
โก! Tokenization is the first thing you do in CS336 (language models from scratch). If Marcel can work his magic for the rest of the LM pipeline, the world will be a better place.
1/ Announcing Vidur, an end-to-end system to generate fine-grained action labels on raw robot and human videos. Vidur leads across all metrics on WGO and EgoExoLearn Bench across temporal segmentation, semantic precision, semantic recall, and end-to-end action-label quality ๐งต
We got CUDA kernels running on @Apple Metal GPUs! ๐โก๏ธ
Thanks to GPT 5.6 Sol, our existing CUDA/HIP-style simulation kernels in OpenFPM now directly execute on Apple Silicon GPUs.
Here is how we got there, and what the first real Dam break benchmark says. ๐งต
We are LIVE! ๐จ The Cloud Technical Series: Agentic AI Edition is kicking off right now!
Tune in to see how industry leaders are moving from AI pilots to enterprise-grade production. Join the livestream here ๐ https://t.co/Z1cuUMZ4kj
Full technical details of AlayaWorld have just been released.
AlayaWorld is a full-stack, open-source video world model that supports 720p, 24 FPS streaming video generation with camera control and text-driven event generation.
https://t.co/OtbAUToIJX
Excited to introduce Nativ ๐
Run frontier open models locally on your Mac. No accounts, no subscriptions, no cloud.
โก Built on mlx-vlm โ multimodal + fastest on Apple Silicon
๐ 100% private โ every token generated on your machine
๐ Plug your coding agents into a local endpoint
๐ Live telemetry โ tokens/sec, memory, usage
๐ 100% open source, MIT licensed, free forever
Your Mac is more capable than you think. Stop renting intelligence.
Download for macOS ๐
https://t.co/JoWC2hLXYA
Github repo ๐
https://t.co/ZXxv58SoWu
Long-running models can solve hard open-ended problems, but their persistence can create safety risks that shorter-horizon evaluations miss.
Weโre sharing what we learned from studying a long-running model, and how those findings are shaping our approach to evaluations, alignment, monitoring, and user control.
https://t.co/yePIzJGsAU