OpenMachine-ai / transformer-tricks
A collection of tricks and tools to speed up transformer models
☆30Updated last week
Alternatives and similar repositories for transformer-tricks:
Users that are interested in transformer-tricks are comparing it to the libraries listed below
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆14Updated last year
- ☆20Updated last year
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Updated last year
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆78Updated 6 months ago
- ☆31Updated 7 months ago
- ☆50Updated 5 months ago
- Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…☆25Updated 9 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated last month
- Collection of autoregressive model implementation☆77Updated 3 weeks ago
- This repo is based on https://github.com/jiaweizzhao/GaLore☆23Updated 4 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Updated last week
- QuIP quantization☆48Updated 10 months ago
- ☆47Updated 5 months ago
- ☆22Updated last year
- ☆66Updated 6 months ago
- Google TPU optimizations for transformers models☆90Updated last week
- Repository containing the SPIN experiments on the DIBT 10k ranked prompts☆24Updated 10 months ago
- From GaLore to WeLore: How Low-Rank Weights Non-uniformly Emerge from Low-Rank Gradients. Ajay Jaiswal, Lu Yin, Zhenyu Zhang, Shiwei Liu,…☆42Updated 6 months ago
- ☆49Updated 10 months ago
- QAmeleon introduces synthetic multilingual QA data using PaLM, a 540B large language model. This dataset was generated by prompt tuning P…☆34Updated last year
- [WIP] Transformer to embed Danbooru labelsets☆13Updated 10 months ago
- My fork os allen AI's OLMo for educational purposes.☆30Updated last month
- Triton Implementation of HyperAttention Algorithm☆46Updated last year
- ☆70Updated 5 months ago
- train with kittens!☆53Updated 3 months ago
- ☆41Updated last year
- Various transformers for FSDP research☆34Updated 2 years ago
- DPO, but faster 🚀☆29Updated last month
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆31Updated 5 months ago