TreeQuest, developed by SakanaAI, is a versatile Python library implementing adaptive tree search algorithms—such as AB‑MCTS—for enhancing inference-time performance of large language models (LLMs). It allows developers to define custom state-generation and scoring functions (e.g., via LLMs), and then efficiently explores possible answer trees during runtime. With support for multi-LLM collaboration, checkpointing, and mixed policies, TreeQuest enables smarter, trial‑and‑error question answering by leveraging both breadth (multiple attempts) and depth (iterative refinement) strategies to find better outputs dynamically
Features
- AB‑MCTS-A and AB‑MCTS‑M algorithms
- Support for multi‑LLM generation strategies
- Custom node-generation and scoring via user functions
- Efficient checkpointing and resume capabilities
- Pythonic, lightweight API for search control
- Built-in utilities for extracting top‑k states
Categories
Artificial IntelligenceLicense
Apache License V2.0Follow TreeQuest
Other Useful Business Software
Build Data Resilience - Take the Assessment Today
Is your recovery strategy as strong as you think? Take this quick self-assessment to check your recovery readiness and gain tailored insights. In only 2 minutes, you'll learn where you fall on the recovery readiness scale.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of TreeQuest!