README.md
January 11, 2026 ยท View on GitHub
Parallel Test-Time Scaling for Latent Reasoning Models
This is the implementation for the paper Parallel Test-Time Scaling for Latent Reasoning Models, enabling efficient exploration of continuous thought spaces through stochastic sampling and reward model-guided search. It provides implementations of two stochastic sampling methods (Monte Carlo Dropout and Additive Gaussian Noise) and a LatentRM for best-of-N and beam search strategies. This repository includes training scripts, evaluation pipelines, and inference code for multiple backbone models including COCONUT, CODI, and CoLaR, evaluated on benchmarks such as GSM8K Test, GSM8K Hard, and MultiArith.
Caution
Legacy Repository: This project is no longer maintained here. Please refer to the latest version at https://github.com/ModalityDance/LatentTTS.