Overview

Epoch AI’s Machine Learning Hardware dataset is a collection of AI accelerators, such as graphics processing units (GPUs) and tensor processing units (TPUs), used to develop and deploy machine learning models in the deep learning era.

This documentation describes the processors included in the dataset, its records, data fields, and definitions, and a changelog and acknowledgements.

The data is available on our website as a visualization or table, and is available for download as a CSV file, 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 data, or suggest hardware that should be added, feel free to contact us at data@epoch.ai.

If this data is useful for you, please cite it as illustrated: