xjdr-alt / simple_transformer
Simple Transformer in Jax
☆136Updated 9 months ago
Alternatives and similar repositories for simple_transformer:
Users that are interested in simple_transformer are comparing it to the libraries listed below
- smol models are fun too☆91Updated 4 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆91Updated 3 weeks ago
- Compiling useful links, papers, benchmarks, ideas, etc.☆41Updated 2 weeks ago
- look how they massacred my boy☆63Updated 5 months ago
- ☆106Updated 3 months ago
- smolLM with Entropix sampler on pytorch☆151Updated 5 months ago
- Just large language models. Hackable, with as little abstraction as possible. Done for my own purposes, feel free to rip.☆44Updated last year
- ☆55Updated 3 weeks ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆49Updated this week
- The history files when recording human interaction while solving ARC tasks☆103Updated this week
- supporting pytorch FSDP for optimizers☆80Updated 3 months ago
- Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆170Updated 8 months ago
- ☆97Updated 5 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆103Updated 4 months ago
- ☆27Updated 8 months ago
- Comprehensive analysis of difference in performance of QLora, Lora, and Full Finetunes.☆82Updated last year
- Model Activity VIsualiser☆109Updated this week
- A MAD laboratory to improve AI architecture designs 🧪☆108Updated 3 months ago
- DeMo: Decoupled Momentum Optimization☆185Updated 4 months ago
- ☆215Updated 8 months ago
- Aidan Bench attempts to measure <big_model_smell> in LLMs.☆288Updated 3 weeks ago
- A really tiny autograd engine☆90Updated 11 months ago
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆63Updated 4 months ago
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆187Updated 10 months ago
- seqax = sequence modeling + JAX☆151Updated 2 weeks ago
- An introduction to LLM Sampling☆77Updated 3 months ago
- ☆53Updated last year
- ☆38Updated 8 months ago
- A puzzle to learn about prompting☆124Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆76Updated 3 weeks ago