Montinger / Transformer-WorkbenchLinks
Playground for Transformers
☆53Updated last year
Alternatives and similar repositories for Transformer-Workbench
Users that are interested in Transformer-Workbench are comparing it to the libraries listed below
Sorting:
- several types of attention modules written in PyTorch for learning purposes☆52Updated last year
- LoRA and DoRA from Scratch Implementations☆215Updated last year
- PyTorch implementation of moe, which stands for mixture of experts☆51Updated 4 years ago
- Efficient Infinite Context Transformers with Infini-attention Pytorch Implementation + QwenMoE Implementation + Training Script + 1M cont…☆83Updated last year
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆102Updated last year
- Unofficial implementation of https://arxiv.org/pdf/2407.14679☆51Updated last year
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆185Updated last year
- A minimal implementation of LLaVA-style VLM with interleaved image & text & video processing ability.☆97Updated 11 months ago
- Implementation of CALM from the paper "LLM Augmented LLMs: Expanding Capabilities through Composition", out of Google Deepmind☆179Updated last year
- minimal GRPO implementation from scratch☆100Updated 9 months ago
- ☆68Updated last year
- 📚 Text Classification with LoRA (Low-Rank Adaptation) of Language Models - Efficiently fine-tune large language models for text classifi…☆55Updated 2 years ago
- Root Mean Square Layer Normalization☆258Updated 2 years ago
- Pytorch Implementation of the paper: "Learning to (Learn at Test Time): RNNs with Expressive Hidden States"☆25Updated last week
- Distributed training (multi-node) of a Transformer model☆89Updated last year
- Implementation of DoRA☆307Updated last year
- This repository contains an implementation of the LLaMA 2 (Large Language Model Meta AI) model, a Generative Pretrained Transformer (GPT)…☆74Updated 2 years ago
- ☆29Updated last year
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆121Updated last year
- Fast instruction tuning with Llama2☆11Updated last year
- a curated list of the role of small models in the LLM era☆110Updated 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 2 years ago
- Building LLaMA 4 MoE from Scratch☆68Updated 7 months ago
- PyTorch implementation of Soft MoE by Google Brain in "From Sparse to Soft Mixtures of Experts" (https://arxiv.org/pdf/2308.00951.pdf)☆79Updated 2 years ago
- ☆42Updated last year
- code for the ddp tutorial☆32Updated 3 years ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆162Updated 8 months ago
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆126Updated 11 months ago
- ☆20Updated 4 years ago
- ☆48Updated last year