DiscoGen: Procedural Generation of Algorithm Discovery Tasks in Machine Learning

This repository contains code for the DiscoGen modular benchmark for automated algorithm discovery.

• Paper: https://arxiv.org/pdf/2603.17863
• GitHub repository: https://github.com/AlexGoldie/discogen/
• Documentation: https://AlexGoldie.github.io/discogen/

Quick Start

Install DiscoGen:

pip install discogen

List available domains:

discogen get-domains

Create a task (the following will include a full implementation, and no editable modules):

discogen create-task --task-domain OnPolicyRL

Create an example task for running an agent (with an incomplete set of modules):

discogen create-task --task-domain OnPolicyRL --example

See the full documentation for detailed usage. Please note that each task_domain has its own set of requirements which may need to be installed, and these are often in conflict with the base DiscoGen requirements. This can be done using the install.sh script provided in each task folder.

Every domain includes references in discogen/domains/<task_domain>/utils/_reference.txt.

Task Domains

Development Setup

1. Set Up Your Development Environment

Install the environment and the pre-commit hooks with:

make install

This will also generate your uv.lock file.

Contributing

We welcome contributions! DiscoGen grows stronger with more tasks and domains.

• Found a bug? Open an issue
• Want to add a task? See our Contributing Guide
• Adding datasets? Check the Dataset Integration Guide

See CONTRIBUTING.md for detailed development guidelines.

Citation

If you use DiscoGen in your research, please cite:

@misc{goldie2026proceduralgenerationalgorithmdiscovery,
  title={Procedural Generation of Algorithm Discovery Tasks in Machine Learning},
  author={Alexander D. Goldie and Zilin Wang and Adrian Hayler and Deepak Nathani and Edan Toledo and Ken Thampiratwong and Aleksandra Kalisz and Michael Beukman and Alistair Letcher and Shashank Reddy and Clarisse Wibault and Theo Wolf and Charles O'Neill and Uljad Berdica and Nicholas Roberts and Saeed Rahmani and Hannah Erlebach and Roberta Raileanu and Shimon Whiteson and Jakob N. Foerster},
  year={2026},
  eprint={2603.17863},
  archivePrefix={arXiv},
  primaryClass={cs.LG},
  url={https://arxiv.org/abs/2603.17863},
}

License

DiscoGen is released under the MIT License.