Arongil / lipschitz-transformersLinks
Don't just regulate gradients like in Muon, regulate the weights too
β27Updated 2 months ago
Alternatives and similar repositories for lipschitz-transformers
Users that are interested in lipschitz-transformers are comparing it to the libraries listed below
Sorting:
- πSmall Batch Size Training for Language Modelsβ62Updated last week
- β33Updated 9 months ago
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token predictionβ71Updated 4 months ago
- β58Updated last year
- β53Updated last year
- Focused on fast experimentation and simplicityβ75Updated 9 months ago
- WIPβ93Updated last year
- Supporting code for the blog post on modular manifolds.β71Updated 2 weeks ago
- β32Updated last year
- A simple, performant and scalable JAX-based world modeling codebaseβ76Updated last week
- Official code for the paper "Attention as a Hypernetwork"β43Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Modelβ55Updated 10 months ago
- Official implementation of the paper "You Do Not Fully Utilize Transformer's Representation Capacity"β31Updated 4 months ago
- β34Updated last year
- research impl of Native Sparse Attention (2502.11089)β61Updated 7 months ago
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion modβ¦β101Updated last month
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorchβ25Updated 8 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT trainingβ132Updated last year
- supporting pytorch FSDP for optimizersβ83Updated 10 months ago
- Code for the paper "Function-Space Learning Rates"β23Updated 4 months ago
- β34Updated last year
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruningβ131Updated 2 weeks ago
- β33Updated 10 months ago
- Explorations into the recently proposed Taylor Series Linear Attentionβ100Updated last year
- β52Updated last year
- β41Updated 6 months ago
- β23Updated last year
- Flash Attention Triton kernel with support for second-order derivativesβ98Updated this week
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]β20Updated last week
- β13Updated 7 months ago