xjdr-alt / simple_transformer
Simple Transformer in Jax
☆136Updated 10 months ago
Alternatives and similar repositories for simple_transformer
Users that are interested in simple_transformer are comparing it to the libraries listed below
Sorting:
- ☆109Updated 4 months ago
- smol models are fun too☆92Updated 6 months ago
- smolLM with Entropix sampler on pytorch☆151Updated 6 months ago
- A really tiny autograd engine☆92Updated last year
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆98Updated 2 months ago
- ☆55Updated 2 months ago
- look how they massacred my boy☆63Updated 6 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
- supporting pytorch FSDP for optimizers☆80Updated 5 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆64Updated 3 weeks ago
- ☆27Updated 10 months ago
- prime-rl is a codebase for decentralized RL training at scale☆89Updated last week
- Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆174Updated 9 months ago
- The history files when recording human interaction while solving ARC tasks☆109Updated 2 weeks ago
- ☆97Updated 7 months ago
- seqax = sequence modeling + JAX☆155Updated last month
- Compiling useful links, papers, benchmarks, ideas, etc.☆46Updated last month
- Gradient descent is cool and all, but what if we could delete it?☆103Updated this week
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆74Updated last week
- A MAD laboratory to improve AI architecture designs 🧪☆115Updated 4 months ago
- Comprehensive analysis of difference in performance of QLora, Lora, and Full Finetunes.☆82Updated last year
- Draw more samples☆189Updated 10 months ago
- A puzzle to learn about prompting☆128Updated 2 years ago
- An introduction to LLM Sampling☆78Updated 4 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆123Updated last year
- ☆38Updated 9 months ago
- could we make an ml stack in 100,000 lines of code?☆42Updated 9 months ago
- ☆217Updated 10 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆50Updated 6 months ago
- This repository contain the simple llama3 implementation in pure jax.☆63Updated 2 months ago