Xuehai Pan

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

Safe-RLHF: Constrained Value Alignment via Safe Reinforcement Learning from Human Feedback

TorchOpt is a high-performance optimizer library built upon PyTorch for easy implementation of functional optimization and gradient-based meta-learning.

OpTree: Optimized PyTree Utilities

Tensors and Dynamic neural networks in Python with strong GPU acceleration

An open source framework that provides a simple, universal API for building distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library.

Automation scripts for setting up a basic development environment.

A collection of LaTeX templates in English/Chinese, with VS Code settings for LaTeX Workshop.

MATE: the Multi-Agent Tracking Environment.

PyTorch Implementation of Soft Actor-Critic Algorithm

No description.

Pytorch domain library for recommendation systems

🍻 Default formulae for the missing package manager for macOS

🍺 The missing package manager for macOS (or Linux)

Cross-platform lib for process and system monitoring in Python

NVIDIA GPUs htop like monitoring tool

The Python programming language

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

📥 Homebrew (un)installer

An Open Source Machine Learning Framework for Everyone

OS-agnostic, system-level binary package manager and ecosystem

A community-maintained mirror of the `nvidia-ml-py` package on PyPI to view the changes easier. PLEASE DO NOT FILE ANY BUG REPORT HERE.

Enabling PyTorch on XLA Devices (e.g. Google TPU)

A VIM-inspired filemanager for the console

LLM inference in C/C++

No description.

Auditing and relabeling cross-distribution Linux wheels.

All Algorithms implemented in Python

A large-scale 7B pretraining language model developed by Baichuan

A parallel framework for population-based multi-agent reinforcement learning.

Auto parallelization for large-scale neural networks

Go Bindings for the NVIDIA Management Library (NVML)

No description.

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

Boost LaTeX typesetting efficiency with preview, compile, autocomplete, colorize, and more.

Collection of library stubs for Python, with static types

Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels

SGLang is a fast serving framework for large language models and vision language models.

Rust bindings for the Python interpreter

Simple, safe way to store and distribute tensors

Development repository for the Triton language and compiler

A high-throughput and memory-efficient inference and serving engine for LLMs

An extremely fast Python package and project manager, written in Rust.

Seamless operability between C++11 and Python

An extremely fast Python linter, written in Rust.

Backported and experimental type hints for Python

📊 A simple command-line utility for querying and monitoring GPU status

Streamlit — A faster way to build and share data apps.

Empowering everyone to build reliable and efficient software.

A conda-smithy repository for nvitop.

A conda-smithy repository for optree.

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

A standard API for reinforcement learning and a diverse set of reference environments (formerly Gym)

A personal project to learn Rust-Python interaction.

The official Python library for the OpenAI API

A conda-smithy repository for nvitop-exporter.

Optional static typing for Python

Renderer for the harmony response format to be used with gpt-oss

A program which ensures source code files have copyright license headers by scanning directory patterns recursively

This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

OmniSafe is a comprehensive and reliable benchmark for safe reinforcement learning.

Source code of the web interface of https://mirrors.tuna.tsinghua.edu.cn/

Flax is a neural network library for JAX that is designed for flexibility.

A place to submit conda recipes before they become fully fledged conda-forge feedstocks

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

Official project repository for the Setuptools build system

Rich is a Python library for rich text and beautiful formatting in the terminal.

DeepEP: an efficient expert-parallel communication library

🤗 smolagents: a barebones library for agents. Agents write python code to call tools and orchestrate other agents.

A Python utility / library to sort imports.

PyTorch native quantization and sparsity for training and inference

A conda-smithy repository for nvidia-ml-py.

Optax is a gradient processing and optimization library for JAX.

An open-source tool-augmented conversational language model from Fudan University

Large Language Model Text Generation Inference

Ongoing research training transformer models at scale

An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and FastChat-T5.

More routines for operating on iterables, beyond itertools

A plugin for Flake8 that provides specializations for type hinting stub files

TensorDict is a pytorch dedicated tensor container.

Safety-Gymnaisum is a highly scalable and customizable safe reinforcement learning environment library.

The Fast Cross-Platform Package Manager

Continuous builder and binary build scripts for pytorch

Software Engineering Course Project

LaTeX template for dissertations in Peking University

No description.

Web Data Mining Course Project

Web Data Mining Course Project

Intro. to Natural Language Processing Course Project

Compiler Design Course Project

Java Programming Course Project

No description.

No description.

No description.

No description.