This repository contains the code to reproduce all experiments in the paper "Wider or Deeper? Scaling LLM Inference-Time Compute with Adaptive Branching Tree Search".

## Environment Setup

For setting up the environment, excluding MLE-Bench, please refer to the instructions in [llm-mcts/README.md](./llm-mcts/README.md#install-requirements).

For detailed instructions on setting up the environment and running experiments for MLE-Bench, please see [mle-bench/README.md](./mle-bench/README.md#competition-installation).

## Running Experiments

All experiment scripts should be run from the `llm-mcts` directory.

### Live Code Bench

To run the Live Code Bench experiment:
```bash
cd llm-mcts
bash run_live_code_bench.sh
```

### Code Contest

To run the Code Contest experiment:
```bash
cd llm-mcts
bash run_code_contest.sh
```

### ARC-AGI

To run the ARC experiment:
```bash
cd llm-mcts
bash run_arc.sh
```

### MLE-Bench

For detailed instructions on setting up the environment and running the agent for MLE-Bench, please refer to [mle-bench/README.md](./mle-bench/README.md#running-the-agent).
