Skip to content
/ mlagility Public

Machine Learning Agility (MLAgility) benchmark and benchmarking tools

License

Notifications You must be signed in to change notification settings

groq/mlagility

Repository files navigation

The MLAgility Project

MLAgility tests onnxflow tests MLAgility GPU tests OS - Linux Made with Python License

MLAgility offers a complementary approach to MLPerf by examining the capability of vendors to provide turnkey solutions to a corpus of hundreds of off-the-shelf models. All of the model scripts and benchmarking code are published as open source software. The performance data is available at our Huggingface Space.

Benchmarking Tool

Our benchit CLI allows you to benchmark Pytorch models without changing a single line of code. The demo below shows BERT-Base being benchmarked on both Nvidia A100 and Intel Xeon. For more information, check out our Tutorials and Tools User Guide.

You can reproduce this demo by trying out the Just Benchmark BERT tutorial.

1000+ Models

Transformers Diffusers popular_on_huggingface torch_hub torchvision timm

This repository is home to a diverse corpus of hundreds of models. We are actively working on increasing the number of models on our model library. You can see the set of models in each category by clicking on the corresponding badge.

Benchmarking results

We are also working on making MLAgility results publicly available at our Huggingface Space. Check it out!

How everything fits together

Architecture block diagram

The diagram above illustrates the MLAgility repository's structure. Simply put, the MLAgility models are leveraged by our benchmarking tool, benchit, to produce benchmarking outcomes showcased on our Hugging Face Spaces page.

Installation

Please refer to our mlagility installation guide to get instructions on how to install mlagility.

Contributing

We are actively seeking collaborators from across the industry. If you would like to contribute to this project, please check out our contribution guide.

License

This project is licensed under the MIT License - see the LICENSE file for details.

About

Machine Learning Agility (MLAgility) benchmark and benchmarking tools

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages