Blogs›Together AI›Autonomous Data Scientist Agent
Autonomous Data Scientist Agent
Together AI enables the creation of autonomous data scientist agents by integrating open-source LLMs with a secure code execution environment (Together Code Interpreter - TCI). These agents leverage the ReAct (Reasoning and Action) pattern, specifically the CodeAct variant, to perform multi-step data analysis tasks. The system allows for flexible prompt engineering to tune agent behavior and supports dynamic installation of necessary libraries within the sandboxed execution environment. This post introduces DSGym, a unified framework that integrates diverse data science evaluation suites behind a single API with standardized abstractions for datasets, agents, and metrics. DSGym unifies and refines existing benchmarks while expanding the scope with novel scientific analysis tasks (90 bioinformatics tasks from academic literature) and challenging end-to-end modeling competitions (92 Kaggle competitions). Beyond evaluation, DSGym provides trajectory generation and synthetic query pipelines for agent training—we demonstrate this by training a 4B model on 2k generated examples, achieving state-of-the-art performance among open-source models. The framework abstracts code execution behind containers, comes with pre-installed dependencies, and offers a unified JSON interface for all benchmarks. DSGym categorizes tasks into Data Analysis and Data Prediction, and introduces DSBio and DSPredict datasets. It also includes a data generation pipeline for agent training, producing 2,000 high-quality query-trajectory pairs for supervised finetuning. The post addresses the 'memorization gap' by filtering out tasks solvable without data access and presents benchmark performance and failure modes, showing that even frontier models underperform on multi-step reasoning, with scientific analysis tasks dominated by domain-grounding errors. DSPredict evaluates the ability of agents to build complete machine learning pipelines, with performance measured by valid submission, median/percentile, and medal achievements.