Liuhong99 / Sophia
The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”
☆957Updated last year
Alternatives and similar repositories for Sophia:
Users that are interested in Sophia are comparing it to the libraries listed below
- Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.☆381Updated 11 months ago
- Fast & Simple repository for pre-training and fine-tuning T5-style models☆1,003Updated 8 months ago
- [NeurIPS 2023] MeZO: Fine-Tuning Language Models with Just Forward Passes. https://arxiv.org/abs/2305.17333☆1,103Updated last year
- Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch☆407Updated 4 months ago
- Convolutions for Sequence Modeling☆878Updated 10 months ago
- Cramming the training of a (BERT-type) language model into limited compute.☆1,331Updated 10 months ago
- Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Pytorch☆641Updated 4 months ago
- Annotated version of the Mamba paper☆483Updated last year
- Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"☆549Updated 4 months ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,128Updated 5 months ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆511Updated 6 months ago
- Language Modeling with the H3 State Space Model☆520Updated last year
- LOMO: LOw-Memory Optimization☆985Updated 10 months ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆672Updated 5 months ago
- Helpful tools and examples for working with flex-attention☆746Updated 3 weeks ago
- maximal update parametrization (µP)☆1,500Updated 9 months ago
- Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate …☆632Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆569Updated this week
- Large Context Attention☆707Updated 3 months ago
- TensorDict is a pytorch dedicated tensor container.☆920Updated this week
- minLoRA: a minimal PyTorch library that allows you to apply LoRA to any PyTorch model.☆458Updated last year
- Ungreedy subword tokenizer and vocabulary trainer for Python, Go & Javascript☆576Updated 10 months ago
- ☆543Updated 4 months ago
- A simple and effective LLM pruning approach.☆741Updated 8 months ago
- Implementation of the conditionally routed attention in the CoLT5 architecture, in Pytorch☆226Updated 8 months ago
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,566Updated last year
- Official code for ReLoRA from the paper Stack More Layers Differently: High-Rank Training Through Low-Rank Updates☆452Updated last year
- Tensors, for human consumption☆1,247Updated 5 months ago
- For optimization algorithm research and development.☆509Updated this week
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆687Updated 8 months ago