OpenMachine-ai / transformer-tricks
A collection of tricks and tools to speed up transformer models
☆32Updated this week
Alternatives and similar repositories for transformer-tricks:
Users that are interested in transformer-tricks are comparing it to the libraries listed below
- ☆22Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 3 weeks ago
- ☆20Updated last year
- This repository contains code for cleaning your training data of benchmark data to help combat data snooping.☆25Updated last year
- ☆49Updated 11 months ago
- Collection of autoregressive model implementation☆81Updated 2 weeks ago
- A library for squeakily cleaning and filtering language datasets.☆46Updated last year
- Utilities for PyTorch distributed☆23Updated this week
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 months ago
- A fast implementation of T5/UL2 in PyTorch using Flash Attention☆82Updated last month
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆84Updated 7 months ago
- ☆58Updated 3 years 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
- some common Huggingface transformers in maximal update parametrization (µP)☆79Updated 2 years ago
- train with kittens!☆53Updated 4 months ago
- ☆21Updated 2 months ago
- ☆42Updated last year
- 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
- HomebrewNLP in JAX flavour for maintable TPU-Training☆48Updated last year
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Updated last year
- [WIP] Transformer to embed Danbooru labelsets☆13Updated 11 months ago
- ☆19Updated 5 months ago
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆81Updated last year
- Utilities for Training Very Large Models☆58Updated 5 months ago
- Repository containing the SPIN experiments on the DIBT 10k ranked prompts☆24Updated 11 months ago
- Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…☆25Updated 10 months ago
- ☆31Updated 8 months ago
- NanoGPT (124M) quality in 2.67B tokens☆28Updated last week
- An implementation of the Llama architecture, to instruct and delight☆21Updated last month
- QuIP quantization☆51Updated 11 months ago