README.md

January 11, 2026 ยท View on GitHub

Parallel Test-Time Scaling for Latent Reasoning Models

Paper HF Models HF Papers Project Logo

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.