pytorch

AI Infra / AI Orchestration / AI Control Plane

A high performance and generic framework for distributed DNN training

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

🧠💬 Articles I wrote about machine learning, archived from MachineCurve.com.

The PyTorch improved version of TPAMI 2017 paper: Face Alignment in Full Pose Range: A 3D Total Solution.

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

Run PyTorch LLMs locally on servers, desktop and mobile

🔥🔥High-Performance Face Recognition Library on PaddlePaddle & PyTorch🔥🔥

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

[CVPR19/TPAMI23] SiamMask: A Framework for Fast Online Object Tracking and Segmentation

Hummingbird compiles trained ML models into tensor computation for faster inference.

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

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

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

Open Machine Learning course

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

[CVPR2020] Adversarial Latent Autoencoders

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.

校招、秋招、春招、实习好项目！带你从零实现一个高性能的深度学习推理库，支持大模型 llama2 、Unet、Yolov5、Resnet等模型的推理。Implement a high-performance deep learning inference library step by step

min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch