Repository Index

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

• nlp
• natural-language-processing
• pytorch
• pytorch-transformers

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

• yolov5
• object-detection
• pytorch
• onnx

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

• deep-learning
• pytorch
• gpu
• machine-learning

Making large AI models cheaper, faster and more accessible

• deep-learning
• hpc
• large-scale
• data-parallelism

VoxCPM2: Tokenizer-Free TTS for Multilingual Speech Generation, Creative Voice Design, and True-to-Life Cloning

• audio
• deeplearning
• minicpm
• python

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

• pytorch
• resnet
• pretrained-models
• pretrained-weights

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

• deep-learning
• machine-learning

LLM training in simple, raw C/CUDA

Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

• ocr
• deep-learning
• crnn
• pytorch

📦 Repomix is a powerful tool that packs your entire repository into a single, AI-friendly file. Perfect for when you need to feed your codebase to Large Language Models (LLMs) or other AI tools like Claude, ChatGPT, DeepSeek, Perplexity, Gemini, Gemma, Llama, Grok, and more.

• ai
• llm
• generative-ai
• chatgpt

See, hear, and interact on your local devices.

• minicpm
• minicpm-v
• multi-modal
• minicpm-o

Open-sourced codes for MiniGPT-4 and MiniGPT-v2 (https://minigpt-4.github.io, https://minigpt-v2.github.io/)

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Multi-lingual large voice generation model, providing inference, training and deployment full-stack ability.

• audio-generation
• gpt-4o
• text-to-speech
• tts

TranslatedPArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice (Core framework of 'PaddlePaddle' (飞桨, Feijiang), for deep learning & machine learning: high-performance single-machine and distributed training, and cross-platform deployment)

• paddlepaddle
• deep-learning
• scalability
• machine-learning

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

• nlp
• pre-trained-model
• unilm
• minilm

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

• deep-learning
• onnx
• neural-networks
• machine-learning

Minimalist ML framework for Rust

Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization pipelines, and OpenAI-compatible/MCP serving.

• pytorch
• speech-recognition
• paraformer
• punctuation

TranslatedOpen-source AI short drama and motion comic creation tool with an infinite canvas, AI agents, storyboarding, image and video generation. Self-hosted and extensible, with a canvas-based approach similar to LibTV and TapNow. Additional details from the Chinese text: Toonflow is a tool for making short dramas and motion comics: an open-source AI video creation platform that integrates script writing, intelligent storyboarding, character and scene asset management, image generation and video generation. It combines an infinite canvas, AI Agents and visual workflows, and supports local deployment, a choice of your own models, cross-platform clients, MCP and plugin extensions.

• ai-content-creation
• generative-ai
• image-generation
• text-to-video

The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement learning and imitation learning.

• reinforcement-learning
• unity3d
• deep-learning
• unity

Toolkit for linearizing PDFs for LLM datasets/training

A WebGL accelerated JavaScript library for training and deploying ML models.

• deep-learning
• typescript
• webgl
• javascript

A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

• gbdt
• gbm
• machine-learning
• data-mining