What problem does it solve?
Automates the repetitive and time-constrained process of proposing, applying, running, and evaluating short ML experiments to improve a small GPT model's validation bits-per-byte metric.
Core Features & Use Cases
- Autonomous Experiment Loop: Reads and modifies train.py, runs training for a fixed five-minute budget, extracts val_bpb and memory usage, and logs results to a structured TSV.
- Crash and Resource Management: Detects crashes, tails logs for quick fixes, enforces a 10-minute hard timeout, and constrains VRAM usage for low-memory GPUs (e.g., RTX 4050).
- Versioned Tracking: Commits experimental changes to git, keeps commits that improve the metric, and rolls back unsuccessful attempts.
- Use Case: Nightly hyperparameter and architecture search on a local workstation to squeeze performance out of constrained GPU hardware.
Quick Start
Start an autoresearch experiment by creating a new branch in the repository and instructing the agent to run a 5-minute training loop, evaluate val_bpb, and log results.