Search awesome repositories

AI Toolkit for Healthcare Imaging

EmotiVoice 😊: a Multi-Voice and Prompt-Controlled TTS Engine

An Extensible Toolkit for Finetuning and Inference of Large Foundation Models. Large Models for All.

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

Code for the paper "Jukebox: A Generative Model for Music"

Background Remover lets you Remove Background from images and video using AI with a simple command line interface that is free and open source.

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages

Easy Docker setup for Stable Diffusion with user-friendly UI

FlashInfer: Kernel Library for LLM Serving

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

A scikit-learn compatible neural network library that wraps PyTorch

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

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.

A programmable Mixture-of-Models router for heterogeneous LLM inference

Fine-tune LLMs from one YAML. Layer streaming trains an 8B model on a 4 GB laptop GPU.

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

Supercharge Your Model Training

Rust bindings for the C++ api of PyTorch.

Probabilistic time series modeling in Python

Implement a reasoning LLM in PyTorch from scratch, step by step

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.

Time series forecasting with PyTorch

An Engine-Agnostic Deep Learning Framework in Java

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

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 的使用，成为一名优秀的深度学习工程师。