learn-cuda. Learn CUDA with PyTorch
345centernet-lightning. Implementation of CenterNet and FairMOT with PyTorch Lightning
69gn-kernels. Python
31quantized-training. Explore training for quantized models
26vision-toolbox. Toolbox for vision tasks. Pre-trained vision backbones on ImageNet with PyTorch Lightning 🚀
17kokoro. https://hf.co/hexgrad/Kokoro-82M
14rocksdb-python. Python bindings for RocksDB. Support Linux, macOS, and Windows
8timm-face. Python
6graph-nn. Minimal and easy-to-understand PyTorch implementations of popular Graph Neural Networks
5gguf-pytorch. Python
5llm-rl. Reinforcement Learning for LLMs
3gan. Learning GANs: WGAN, Relativistic GAN, StyleGAN, pix2pix...
3Triton-Puzzles. Jupyter Notebook
3lmdb-python. Another Python binding for LMDB. Built with Cython
2word2vec-numba. Python
2gau-nernst.github.io. HTML
2ray-tracing-rust. Ray Tracing in One Weekend - Rust Edition
2learn-cpu. C++
2gemma3-int4. PyTorch inference library for Gemma 3 INT4
2llm. Python
2lectures. Material for gpu-mode lectures
1torchtune. A Native-PyTorch Library for LLM Fine-tuning
1pytorch-models. Collection of image, text, and audio models: ViT, BERT, Whisper...
1MOOC. Jupyter Notebook
1scl-2021-ds. Address Elements Extraction with transformers
1CodeComps. Data Science projects and competitions done by me
1