Repository Index

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

• machine-learning
• models
• ui
• ui-components

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

TranslatedChatGLM-6B: An Open Bilingual Dialogue Language Model | Open-source bilingual dialogue language model

[EMNLP2025] LightRAG: Simple and Fast Retrieval-Augmented Generation

• knowledge-graph
• large-language-models
• retrieval-augmented-generation
• genai

A generative speech model for daily dialogue.

• agent
• text-to-speech
• chat
• chatgpt

An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena.

A Python library for extracting structured information from unstructured text using LLMs with precise source grounding and interactive visualization.

• llm
• nlp
• python
• gemini-ai

Easily train a good VC model with voice data <= 10 mins!

• change
• sovits
• vits
• voice

DSPy: The framework for programming—not prompting—language models

Training-free language model modification: Remove censorship, suppress slop, reduce sycophancy, and more

• abliteration
• llm
• transformer
• censorship-resistance

SGLang is a high-performance serving framework for large language models and multimodal models.

• cuda
• inference
• llama
• llm

Instant voice cloning by MIT and MyShell. Audio foundation model.

• text-to-speech
• tts
• voice-clone
• zero-shot-tts

OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamically to do so.

• chatgpt
• language-model
• rlhf
• ai

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

Official Code for DragGAN (SIGGRAPH 2023)

• artificial-intelligence
• generative-adversarial-network
• generative-models
• image-manipulation

A simple Python Pydantic model for Honkai: Star Rail parsed data from the Mihomo API.

• honkai-star-rail
• mihomo
• python
• star-rail-api

Build and run agents you can see, understand and trust.

• agent
• chatbot
• large-language-models
• llm

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.

Non-autoregressive System 1 decision engine. Typed choice, score and yes/no decisions over any text in a single forward pass, in 100+ languages, with a router that picks the right checkpoint per request.

• calibration
• classification
• decision-model
• huggingface

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

• deep-learning
• diffusion
• image-generation
• pytorch

Let us control diffusion models!

Machine Learning From Scratch. Bare bones NumPy implementations of machine learning models and algorithms with a focus on accessibility. Aims to cover everything from linear regression to deep learning.

• machine-learning
• deep-learning
• deep-reinforcement-learning
• machine-learning-from-scratch