Niccolo-Ajroldi / plainLMLinks
Minimal pretraining script for language modeling in PyTorch. Supporting torch compilation and DDP. It includes a model implementation and a data preprocessing.
☆32Updated this week
Alternatives and similar repositories for plainLM
Users that are interested in plainLM are comparing it to the libraries listed below
Sorting:
- nanoGPT-like codebase for LLM training☆102Updated 2 months ago
- ☆17Updated last year
- supporting pytorch FSDP for optimizers☆84Updated 8 months ago
- ☆33Updated last month
- ☆206Updated 8 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated 2 weeks ago
- ☆53Updated 10 months ago
- ☆232Updated 5 months ago
- A library for unit scaling in PyTorch☆128Updated last month
- ☆70Updated 8 months ago
- ☆11Updated 3 months ago
- ☆53Updated last year
- Accelerated First Order Parallel Associative Scan☆184Updated 11 months ago
- ☆11Updated 10 months ago
- 🧱 Modula software package☆216Updated 2 weeks ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆81Updated 9 months ago
- 📄Small Batch Size Training for Language Models☆41Updated this week
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆275Updated 3 weeks ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆63Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆149Updated last month
- A MAD laboratory to improve AI architecture designs 🧪☆123Updated 7 months ago
- ☆23Updated 2 years ago
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆60Updated 5 months ago
- Parallel Associative Scan for Language Models☆18Updated last year
- Distributed K-FAC preconditioner for PyTorch☆89Updated this week
- ☆13Updated 6 months ago
- Lightweight torch implementation of rigl, a sparse-to-sparse optimizer.☆57Updated 3 years ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆14Updated last year
- Efficient Riemannian Optimization on Stiefel Manifold via Cayley Transform☆42Updated 6 years ago