mgrankin / minGPT
minGPT in JAX
☆46Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for minGPT
- A port of muP to JAX/Haiku☆25Updated 2 years ago
- LoRA for arbitrary JAX models and functions☆133Updated 8 months ago
- ☆16Updated 2 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆128Updated last month
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆30Updated 11 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆79Updated 9 months ago
- A functional training loops library for JAX☆85Updated 9 months ago
- If it quacks like a tensor...☆52Updated last week
- ☆40Updated 4 months ago
- Neural Networks for JAX