clu0 / unet.cuLinks
UNet diffusion model in pure CUDA
☆655Updated last year
Alternatives and similar repositories for unet.cu
Users that are interested in unet.cu are comparing it to the libraries listed below
Sorting:
- Implementation of Diffusion Transformer (DiT) in JAX☆298Updated last year
- Alex Krizhevsky's original code from Google Code☆197Updated 9 years ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆585Updated 4 months ago
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆438Updated 9 months ago
- Learnings and programs related to CUDA☆428Updated 5 months ago
- Simple Byte pair Encoding mechanism used for tokenization process . written purely in C☆139Updated last year
- Simple MPI implementation for prototyping or learning☆292Updated 4 months ago
- Best practices & guides on how to write distributed pytorch training code☆552Updated last month
- GPU programming related news and material links☆1,860Updated 2 months ago
- ☆177Updated last year
- Puzzles for learning Triton☆2,170Updated last year
- ☆532Updated 4 months ago
- (WIP) A small but powerful, homemade PyTorch from scratch.☆661Updated this week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆1,917Updated 3 months ago
- For optimization algorithm research and development.☆552Updated this week
- 100 days of building GPU kernels!☆547Updated 7 months ago
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆828Updated 4 months ago
- Annotated version of the Mamba paper☆491Updated last year
- Tutorials on tinygrad☆444Updated 2 months ago
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,429Updated this week
- Complete solutions to the Programming Massively Parallel Processors Edition 4☆602Updated 5 months ago
- Simple and readable code for training and sampling from diffusion models☆666Updated 6 months ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆196Updated 6 months ago
- ☆547Updated last year
- What would you do with 1000 H100s...☆1,133Updated last year
- An ML Systems Onboarding list☆953Updated 10 months ago
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆196Updated 2 years ago
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,540Updated 11 months ago
- Following Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆172Updated last year
- Puzzles for exploring transformers☆380Updated 2 years ago