lweitkamp's Stars
richzhang/PerceptualSimilarity
LPIPS metric. pip install lpips
lucidrains/vector-quantize-pytorch
Vector (and Scalar) Quantization, in Pytorch
google/grain
somaniarushi/code_mcts
👩🏻💻 LLM + MCTS for Humaneval
karpathy/LLM101n
LLM101n: Let's build a Storyteller
google-deepmind/nanodo
cuda-mode/triton-index
Cataloging released Triton kernels.
srush/Triton-Puzzles
Puzzles for learning Triton
microsoft/mscclpp
MSCCL++: A GPU-driven communication stack for scalable AI applications
Deep-Learning-Profiling-Tools/triton-viz
nod-ai/techtalks
facebookresearch/xformers
Hackable and optimized Transformers building blocks, supporting a composable construction.
triton-lang/triton
Development repository for the Triton language and compiler
Cjkkkk/CUDA_gemm
A simple high performance CUDA GEMM implementation.
forhaoliu/ringattention
Transformers with Arbitrarily Large Context
beartype/beartype
Unbearably fast near-real-time hybrid runtime-static type-checking in pure Python.
siboehm/ShallowSpeed
Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
lucidrains/autoregressive-linear-attention-cuda
CUDA implementation of autoregressive linear attention, with all the latest research findings
olcf/cuda-training-series
Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)
wangzyon/NVIDIA_SGEMM_PRACTICE
Step-by-step optimization of CUDA SGEMM
dqxiu/ICL_PaperList
Paper List for In-context Learning 🌷
KernelTuner/kernel_tuner_tutorial
A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/
google-research/tuning_playbook
A playbook for systematically maximizing the performance of deep learning models.
lucidrains/medical-chatgpt
Implementation of ChatGPT, but tailored towards primary care medicine, with the reward being able to collect patient histories in a thorough and efficient manner and come up with a reasonable differential diagnosis
infocusp/diffusion_models
Minimal standalone example of diffusion model
ElisevanderPol/PRAE
Plannable Approximations to MDP Homomorphisms: Equivariance under Actions