https://github.com/autogluon/autogluon-assistant

ML Assistant for Competitive Machine Learning
https://github.com/autogluon/autogluon-assistant

automl data-science llms machine-learning

Last synced: 21 days ago
JSON representation

ML Assistant for Competitive Machine Learning

• Host: GitHub
• URL: https://github.com/autogluon/autogluon-assistant
• Owner: autogluon
• License: apache-2.0
• Created: 2024-04-29T07:23:46.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2025-01-06T23:43:53.000Z (6 months ago)
• Last Synced: 2025-01-20T09:07:05.924Z (5 months ago)
• Topics: automl, data-science, llms, machine-learning
• Language: Python
• Homepage:
• Size: 1.8 MB
• Stars: 98
• Watchers: 7
• Forks: 12
• Open Issues: 14
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE
  • Code of conduct: CODE_OF_CONDUCT.md

Awesome Lists containing this project

README

        


# AutoGluon Assistant (aka MLZero)

[![Python Versions](https://img.shields.io/badge/python-3.8%20%7C%203.9%20%7C%203.10%20%7C%203.11-blue)](https://pypi.org/project/autogluon.assistant/)

[![GitHub license](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](./LICENSE)

[![Continuous Integration](https://github.com/autogluon/autogluon-assistant/actions/workflows/continuous_integration.yml/badge.svg)](https://github.com/autogluon/autogluon-assistant/actions/workflows/continuous_integration.yml)

[![Project Page](https://img.shields.io/badge/Project_Page-MLZero-blue)](https://project-mlzero.github.io/)



AutoGluon Assistant (aka MLZero) is a multi-agent system that automates end-to-end multimodal machine learning or deep learning workflows by transforming raw multimodal data into high-quality ML solutions with zero human intervention. Leveraging specialized perception agents, dual-memory modules, and iterative code generation, it handles diverse data formats while maintaining high success rates across complex ML tasks.

## 💾 Installation

AutoGluon Assistant is supported on Python 3.8 - 3.11 and is available on Linux, MacOS, and Windows.

You can install from source (new version will be released to PyPI soon):

```bash

pip install uv

uv pip install git+https://github.com/autogluon/autogluon-assistant.git

```

## Quick Start

For detailed usage instructions, OpenAI/Azure setup, and advanced configuration options, see our [Getting Started Tutorial](docs/tutorials/getting_started.md).

### API Setup

MLZero uses AWS Bedrock by default. Configure your AWS credentials:

```bash

export AWS_DEFAULT_REGION=""

export AWS_ACCESS_KEY_ID=""

export AWS_SECRET_ACCESS_KEY=""

```

We also support OpenAI. More LLM providers' support (e.g. Anthropic, Azure, etc.) will be added soon.

### Basic Usage

![Demo](https://github.com/autogluon/autogluon-assistant/blob/main/docs/assets/cli_demo.gif)

```bash

mlzero -i  [-u ]

```

## Citation

If you use Autogluon Assistant (MLZero) in your research, please cite our paper:

```bibtex

@misc{fang2025mlzeromultiagentendtoendmachine,

      title={MLZero: A Multi-Agent System for End-to-end Machine Learning Automation}, 

      author={Haoyang Fang and Boran Han and Nick Erickson and Xiyuan Zhang and Su Zhou and Anirudh Dagar and Jiani Zhang and Ali Caner Turkmen and Cuixiong Hu and Huzefa Rangwala and Ying Nian Wu and Bernie Wang and George Karypis},

      year={2025},

      eprint={2505.13941},

      archivePrefix={arXiv},

      primaryClass={cs.MA},

      url={https://arxiv.org/abs/2505.13941}, 

}

```