Thien Tran

learn-cuda. Learn CUDA with PyTorch

centernet-lightning. Implementation of CenterNet and FairMOT with PyTorch Lightning

gn-kernels. Python

quantized-training. Explore training for quantized models

vision-toolbox. Toolbox for vision tasks. Pre-trained vision backbones on ImageNet with PyTorch Lightning 🚀

kokoro. https://hf.co/hexgrad/Kokoro-82M

rocksdb-python. Python bindings for RocksDB. Support Linux, macOS, and Windows

timm-face. Python

graph-nn. Minimal and easy-to-understand PyTorch implementations of popular Graph Neural Networks

gguf-pytorch. Python

llm-rl. Reinforcement Learning for LLMs

gan. Learning GANs: WGAN, Relativistic GAN, StyleGAN, pix2pix...

Triton-Puzzles. Jupyter Notebook

lmdb-python. Another Python binding for LMDB. Built with Cython

word2vec-numba. Python

gau-nernst.github.io. HTML

ray-tracing-rust. Ray Tracing in One Weekend - Rust Edition

learn-cpu. C++

gemma3-int4. PyTorch inference library for Gemma 3 INT4

llm. Python

flow-matching. Python

lectures. Material for gpu-mode lectures

torchtune. A Native-PyTorch Library for LLM Fine-tuning

pytorch-models. Collection of image, text, and audio models: ViT, BERT, Whisper...

MOOC. Jupyter Notebook

scl-2021-ds. Address Elements Extraction with transformers

CodeComps. Data Science projects and competitions done by me