Zephon brings elastically deterministic data loading to TorchTitan with identical batches across GPU counts

josh_wills · x · 2026-10-17

PyTorch's blog details Zephon's integration with TorchTitan: a 1B-param model trained on 16 GPUs resumed from step 150 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 costly preprocessed corpus copies. Determinism matters more now that agents run experiments in parallel and can't notice when results depend on GPU count rather than the change being tested. Benchmarks show 22.4k-26.6k usable tokens/sec per GPU versus 1.1k-9.2k for standard loaders. Losses can still differ slightly due to floating-point reduction order.

Related event: Zephon Integrates with TorchTitan for Deterministic Recovery Across GPU Counts(2 posts)→

Original post →

More from Infra

Infra channel →