mlcommons/training

Reference implementations of MLPerf® inference benchmarks

MLE-bench is a benchmark for measuring how well AI agents perform at machine learning engineering

Models and examples built with TensorFlow

Because `model.fit()` isn't an explanation

Example models using DeepSpeed

Ongoing research training transformer models at scale

An agentic Machine Learning Engineer

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

Code, labs, and resources for O'Reilly AI Systems Performance Engineering: GPU optimization, distributed training, inference scaling, and full-stack tuning.