gau-nernst / learn-cuda
Learn CUDA with PyTorch
☆20Updated 3 months ago
Alternatives and similar repositories for learn-cuda:
Users that are interested in learn-cuda are comparing it to the libraries listed below
- ☆21Updated 2 months ago
- Make triton easier☆47Updated 10 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆55Updated this week
- Custom triton kernels for training Karpathy's nanoGPT.☆18Updated 6 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- See https://github.com/cuda-mode/triton-index/ instead!☆11Updated 11 months ago
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆65Updated last month
- ☆20Updated last year
- Custom kernels in Triton language for accelerating LLMs☆18Updated last year
- Write a fast kernel and run it on Discord. See how you compare against the best!☆44Updated this week
- NanoGPT-speedrunning for the poor T4 enjoyers☆63Updated last week
- ☆20Updated last year
- ML/DL Math and Method notes☆60Updated last year
- ☆78Updated 10 months ago
- Various transformers for FSDP research☆37Updated 2 years ago
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆82Updated last year
- prime-rl is a codebase for decentralized RL training at scale☆85Updated this week
- ☆17Updated last year
- Load compute kernels from the Hub☆115Updated last week
- An implementation of the Llama architecture, to instruct and delight☆21Updated 3 months ago
- ☆47Updated 8 months ago
- Collection of autoregressive model implementation☆85Updated last week
- ☆14Updated 9 months ago
- making the official triton tutorials actually comprehensible☆27Updated last month
- ☆43Updated last year
- ☆27Updated 9 months ago
- JAX implementation of the Mistral 7b v0.2 model☆35Updated 10 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆130Updated last year
- Cray-LM unified training and inference stack.☆22Updated 3 months ago
- Simple repository for training small reasoning models☆27Updated 2 months ago