Overview

Epoch’s AI Models Dataset is a collection of machine learning models useful for research about trends in the history and future of artificial intelligence. It includes over two thousand machine learning models, encompassing a broad range of domains and scales.

This documentation describes which models are contained within the database, its records (including data fields and definitions), processes for adding new entries and auditing accuracy. It also includes a changelog and acknowledgements.

The data contain several subsets, which can be viewed using the interactive visualization. Notable AI models are those that meet our notability criteria. Frontier models are models that were in the top 10 of training compute as of the time of their release. Large-scale models are models trained with at least 10²³ floating-point operations. All models shows every model in the dataset, including models that do not qualify for the above categories, such as models found during our research on algorithmic progress, AI for biology, or other investigations.

The dataset is available on our website as a visualization or table, and is available for download in CSV format, updated daily. For a quick-start example of loading the data and working with it in your research, see this Google Colab demo notebook.

If you would like to ask any questions about the database, or suggest a model that should be added, contact us at data@epoch.ai.

If this dataset is useful for you, please cite it.