ulrichstern / cuda-convnet
Alex Krizhevsky's original code from Google Code
☆190Updated 8 years ago
Related projects ⓘ
Alternatives and complementary repositories for cuda-convnet
- UNet diffusion model in pure CUDA☆584Updated 4 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆483Updated 3 weeks ago
- An ML Systems Onboarding list☆545Updated this week
- A really tiny autograd engine☆87Updated 7 months ago
- The Tensor (or Array)☆411Updated 3 months ago
- High-Performance FP32 Matrix Multiplication on CPU☆301Updated this week
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆157Updated last year
- ☆133Updated 9 months ago
- The Autograd Engine☆534Updated 2 months ago
- Simple Byte pair Encoding mechanism used for tokenization process . written purely in C☆120Updated last week
- Recreating PyTorch from scratch (C/C++, CUDA, NCCL and Python, with multi-GPU support and automatic differentiation!)☆114Updated 5 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆107Updated last year
- The Multilayer Perceptron Language Model☆523Updated 3 months ago
- Best practices & guides on how to write distributed pytorch training code☆286Updated 2 weeks ago
- Implementation of Diffusion Transformer (DiT) in JAX☆252Updated 5 months ago
- ☆82Updated 8 months ago
- Solve puzzles to improve your tinygrad skills!☆87Updated 2 months ago
- Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆167Updated 3 months ago
- ☆152Updated this week
- Solve puzzles. Learn CUDA.☆61Updated 11 months ago
- Fast bare-bones BPE for modern tokenizer training☆142Updated last month
- Tutorials on tinygrad☆180Updated last week
- Puzzles for learning Triton☆1,135Updated this week
- For optimization algorithm research and development.☆449Updated this week
- Tensor library with autograd using only Rust's standard library☆62Updated 4 months ago
- ring-attention experiments☆97Updated last month
- pytorch from scratch in pure C/CUDA and python☆37Updated last month
- Flash Attention in ~100 lines of CUDA (forward pass only)☆626Updated 7 months ago
- Annotated version of the Mamba paper☆457Updated 8 months ago
- LLM training in simple, raw C/CUDA☆86Updated 6 months ago