AkiRusProd / numpy-transformerLinks
A numpy implementation of the Transformer model in "Attention is All You Need"
☆55Updated 11 months ago
Alternatives and similar repositories for numpy-transformer
Users that are interested in numpy-transformer are comparing it to the libraries listed below
Sorting:
- Notes on quantization in neural networks☆86Updated last year
- ☆174Updated 5 months ago
- LLaMA 2 implemented from scratch in PyTorch☆336Updated last year
- An extension of the nanoGPT repository for training small MOE models.☆152Updated 3 months ago
- Prune transformer layers☆69Updated last year
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆109Updated last year
- ☆69Updated last year
- Distributed training (multi-node) of a Transformer model☆72Updated last year
- ☆193Updated 4 months ago
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆51Updated last year
- ☆159Updated last year
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆106Updated last year
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆92Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆185Updated 3 weeks ago
- This repository contains an implementation of the LLaMA 2 (Large Language Model Meta AI) model, a Generative Pretrained Transformer (GPT)…☆68Updated last year
- Simple Adaptation of BitNet☆32Updated last year
- Implementation of Infini-Transformer in Pytorch☆111Updated 5 months ago
- ☆39Updated last month
- Code for studying the super weight in LLM☆107Updated 6 months ago
- Mixed precision training from scratch with Tensors and CUDA☆24Updated last year
- Unofficial implementation of https://arxiv.org/pdf/2407.14679☆45Updated 9 months ago
- Training small GPT-2 style models using Kolmogorov-Arnold networks.☆119Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆201Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆556Updated last week
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆252Updated 3 months ago
- several types of attention modules written in PyTorch for learning purposes☆54Updated 8 months ago
- ML/DL Math and Method notes☆61Updated last year
- Experiments on Multi-Head Latent Attention☆92Updated 10 months ago
- Simple Byte pair Encoding mechanism used for tokenization process . written purely in C☆133Updated 7 months ago