AkiRusProd / numpy-transformerLinks
A numpy implementation of the Transformer model in "Attention is All You Need"
☆58Updated last year
Alternatives and similar repositories for numpy-transformer
Users that are interested in numpy-transformer are comparing it to the libraries listed below
Sorting:
- LLaMA 2 implemented from scratch in PyTorch☆353Updated 2 years ago
- Notes on quantization in neural networks☆104Updated last year
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆116Updated 2 years ago
- Well documented, unit tested, type checked and formatted implementation of a vanilla transformer - for educational purposes.☆262Updated last year
- ☆206Updated 9 months ago
- ☆173Updated last year
- Recreating PyTorch from scratch (C/C++, CUDA, NCCL and Python, with multi-GPU support and automatic differentiation!)☆159Updated last year
- Custom kernels in Triton language for accelerating LLMs☆26Updated last year
- Best practices & guides on how to write distributed pytorch training code☆487Updated 7 months ago
- An extension of the nanoGPT repository for training small MOE models.☆196Updated 7 months ago
- ☆222Updated last week
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆194Updated 4 months ago
- a minimal cache manager for PagedAttention, on top of llama3.☆123Updated last year
- ☆13Updated 11 months ago
- Annotated version of the Mamba paper☆489Updated last year
- Distributed training (multi-node) of a Transformer model☆84Updated last year
- Cataloging released Triton kernels.☆261Updated last month
- Simple Byte pair Encoding mechanism used for tokenization process . written purely in C☆136Updated 10 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆577Updated last month
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆108Updated 2 years ago
- LoRA and DoRA from Scratch Implementations☆211Updated last year
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆122Updated 11 months ago
- For optimization algorithm research and development.☆539Updated 2 weeks ago
- A Simplified PyTorch Implementation of Vision Transformer (ViT)☆211Updated last year
- ☆45Updated 4 months ago
- Learn CUDA with PyTorch☆85Updated 2 weeks ago
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆92Updated 2 years ago
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆190Updated 2 years ago
- Tutorial for how to build BERT from scratch☆99Updated last year
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆52Updated last year