Models and Pre-Trained Weights

Models Repository

The models repository contains definitions of different deep learning models with examples for different domains like computer vision and natural language processing.

This includes image classification models such as MobileNetV2, SqueezeNet and ResNet, object detection models such as YOLOX and language models like BERT and RoBERTa.

Be sure to check out the up-to-date collection of models to get you started. Pre-trained weights are available for every supported architecture in this collection. You will also find a spotlight of community contributed models.

Burn-LM (alpha)

Burn-LM is an LLM inference engine built on Burn. It provides access to large language models with open-source pre-trained weights and supports running, fine-tuning, and experimenting with them on any Burn backend.

Unlike tools focused solely on inference, Burn-LM is designed to work in a unified way across different models and tasks, making it easier to explore both inference and training workflows within the same framework.