Stars
🤗 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.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
The simplest, fastest repository for training/finetuning medium-sized GPTs.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Deep Learning papers reading roadmap for anyone who are eager to learn this amazing tech!
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.
[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.
Stable Diffusion with Core ML on Apple Silicon
Datasets, Transforms and Models specific to Computer Vision
A reactive notebook for Python — run reproducible experiments, query with SQL, execute as a script, deploy as an app, and version with git. Stored as pure Python. All in a modern, AI-native editor.
A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
DALL·E Mini - Generate images from a text prompt
Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.
20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
Minimal and clean examples of machine learning algorithms implementations
Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stream data in real-time to PyTorch/TensorFlow. https://activelo…
An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.
Pythonic AI generation of images and videos
Fast, correct Python JSON library supporting dataclasses, datetimes, and numpy
pipreqs - Generate pip requirements.txt file based on imports of any project. Looking for maintainers to move this project forward.
Official Implementation of OCR-free Document Understanding Transformer (Donut) and Synthetic Document Generator (SynthDoG), ECCV 2022
ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution