v0.1.0
First PyPI release of ThinkBooster — a framework for test-time compute scaling of LLM reasoning.
Features
- Strategies: Baseline, Self-Consistency (majority voting), Best-of-N, Beam Search, Adaptive Scaling
- Backends: vLLM (primary), HuggingFace Transformers, OpenAI-compatible APIs
- Scorers: Entropy, Perplexity, PRM, Sequence Probability, UHead
- Evaluation: Exact match with LaTeX/symbolic normalization, LLM judge
- Datasets: MATH-500, GSM8K, OlympiadBench, GaoKao, Minerva Math, AIME, HumanEval+, MBPP+
- Extras: Visual debugger for reasoning traces, OpenAI-compatible API endpoint
Install
pip install thinkbooster