Search awesome repositories

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

A unified, comprehensive and efficient recommendation library

LLM training code for Databricks foundation models

Open-source evaluation toolkit of large multi-modality models (LMMs), support 220+ LMMs, 80+ benchmarks

🚀 An open-source, hands-on curriculum bridging the gap from basic RL concepts to LLM alignment, RLVR, and advanced Agentic systems.

Scalable and user friendly neural :brain: forecasting algorithms.

⚡️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.

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

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

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 highly efficient implementation of Gaussian Processes in PyTorch

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

The official CLI and Python client for the Hugging Face Hub.

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

AI Infra / AI Orchestration / AI Control Plane

Fine-tuning ChatGLM-6B with PEFT | 基于 PEFT 的高效 ChatGLM 微调

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

Run PyTorch LLMs locally on servers, desktop and mobile

Models, data loaders and abstractions for language processing, powered by PyTorch

Fast, flexible and easy to use probabilistic modelling in Python.

🛠 All-in-one web-based IDE specialized for machine learning and data science.

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

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

GeoAI: Artificial Intelligence for Geospatial Data

Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2

Accelerated deep learning R&D

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