标签：PyTorch

Accessible large language models via k-bit quantization for PyTorch.

BertViz: Visualize Attention in Transformer Models

A framework for efficient model inference with omni-modality models

a cross-platform image super-resolution tool

SD.Next: All-in-one WebUI for AI generative image and video creation, captioning and processing

FlashInfer: Kernel Library for LLM Serving

An open, programmable decision layer for models and compute.

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning. Ongoing development and maintenance by t2k.

Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

Fault-tolerant, highly scalable GPU orchestration, and a machine learning framework designed for training models with billions to trillions of parameters

Setup and customize deep learning environment in seconds.

Standardized Distributed Generative and Predictive AI Inference Platform for Scalable, Multi-Framework Deployment on Kubernetes

Silero Models: pre-trained text-to-speech models made embarrassingly simple

Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

Dockerized OpenAI-compatible wrapper for Kokoro-82M text-to-speech w/multiplatform CPU, AMD, NVIDIA GPU PyTorch; multi-speaker, clone-tuning, caption timestamps, SSML, websockets, optional readalong web UI

ZenML 🙏: One AI Platform from Pipelines to Agents. https://zenml.io.

Framework agnostic sliced/tiled inference + interactive ui + error analysis plots

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡

Superduper: End-to-end framework for building custom AI applications and agents.

深度学习辅助漫画翻译工具, 支持一键机翻和简单的图像/文本编辑 | Yet another computer-aided comic/manga translation tool powered by deeplearning

Production First and Production Ready End-to-End Speech Recognition Toolkit

A PyTorch Library for Accelerating 3D Deep Learning Research

cube studio开源云原生一站式机器学习/深度学习/大模型AI平台，mlops算法链路全流程，算力租赁平台，notebook在线开发，拖拉拽任务流pipeline编排，多机多卡分布式训练，超参搜索，推理服务VGPU虚拟化，边缘计算，标注平台自动化标注，deepseek等大模型sft微调/奖励模型/强化学习训练，vllm/ollama/mindie大模型多机推理，私有知识库，AI模型市场，支持国产cpu/gpu/npu 昇腾生态，支持RDMA，支持pytorch/tf/mxnet/deepspeed/paddle/colossalai/horovod/ray/volcano等分布式

Machine Learning and Agentic AI Resources, Practice and Research

An Open Source text-to-speech system built by inverting Whisper.

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

《Pytorch实用教程》（第二版）无论是零基础入门，还是CV、NLP、LLM项目应用，或是进阶工程化部署落地，在这里都有。相信在本书的帮助下，读者将能够轻松掌握 PyTorch 的使用，成为一名优秀的深度学习工程师。

Lightning fast C++/CUDA neural network framework

[EMNLP 2022] An Open Toolkit for Knowledge Graph Extraction and Construction

Simplify your onnx model

⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

Serve, optimize and scale PyTorch models in production

中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard

Build a modern LLM from scratch. Every line commented. Explained like we are five.

Fengshenbang-LM(封神榜大模型)是IDEA研究院认知计算与自然语言研究中心主导的大模型开源体系，成为中文AIGC和认知智能的基础设施。

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and models from research to production.

A retargetable MLIR-based machine learning compiler and runtime toolkit.

A simple, high-quality voice conversion tool focused on ease of use and performance.

中文nlp解决方案(大模型、数据、模型、训练、推理)

Multi-LoRA inference server that scales to 1000s of fine-tuned LLMs

A python library for self-supervised learning and latent world models on images.

A collection of scientific methods, processes, algorithms, and systems to build stories & models.

AI Infra / AI Orchestration / AI Control Plane

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

GeoAI: Artificial Intelligence for Geospatial Data

Accelerated deep learning R&D