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
Propel Software: Product Value Management Platform for Manufacturers
Propel is a cloud-native Product Value Management platform that unifies PLM, QMS, and PIM in one connected system, giving manufacturers complete visibility and control across the entire product lifecycle. It provides a single source of truth for all product data, streamlines change management, strengthens quality and compliance processes, and accelerates time-to-market by eliminating the silos and manual steps that slow teams down.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of TreeQuest!