《动手学深度学习》：面向中文读者、能运行、可讨论。中英文版被70多个国家的500多所大学用于教学。

Machine learning, in numpy

An educational resource to help anyone learn deep reinforcement learning.

Nano vLLM

Ring attention implementation with flash attention

BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.

A Quirky Assortment of CuTe Kernels

Applied AI experiments and examples for PyTorch

SIMPLE DMRG tutorial, created for the Trieste summer school on quantum spin liquids.

Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).

Sampling with gradient-based Markov Chain Monte Carlo approaches

Assembler and Decompiler for NVIDIA (Maxwell Pascal Volta Turing Ampere) GPUs.

Implementation of ConjugateGradients method using C and Nvidia CUDA

Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".

An implementation of the transformer quantum state, a multi-purpose model for quantum many-body problems

A more "sophisticated" DMRG program, based on simple-dmrg

Mixed precision training from scratch with Tensors and CUDA

Code for 'Unbiased Monte Carlo Cluster Updates with Autoregressive Neural Networks'.

Jack's Jax Utilities

A demonstration code to show semi-empirical preconditioning Davidson Algorithm

Variational Monte Carlo solver written in Python