Burn alternatives for AI agents.

Open Machine Learning Compiler Framework

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.

Rust bindings for the C++ api of PyTorch.

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Deep Learning for humans

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

Deepfakes Software For All

12 Weeks, 24 Lessons, AI for All!

Cross-platform, customizable ML solutions for live and streaming media.

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

An Open Source Machine Learning Framework for Everyone

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠