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
Our Free Plans just got better! | Auth0
You asked, we delivered! Auth0 is excited to expand our Free and Paid plans to include more options so you can focus on building, deploying, and scaling applications without having to worry about your security. Auth0 now, thank yourself later.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of TreeQuest!