Sigpy alternatives for AI agents.

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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Open Machine Learning Compiler Framework

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

The fastai deep learning library

🤗 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.

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

We write your reusable computer vision tools. 💜

Open3D: A Modern Library for 3D Data Processing

🧑‍🏫 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, ... 🧠

Deepfakes Software For All

💫 Industrial-strength Natural Language Processing (NLP) in Python