Train a language model in your browser with WebGPU!
I built a playground for training sequence models (Transformers, LSTMs, GRUs, vanilla RNNs) completely in your browser on synthetic tasks like sorting and simple natural language datasets like TinyStories. You can fiddle with 50 experiment knobs to build your own model, which can be as big as you have the VRAM to accommodate.
You don't have to install anything—all you need is a browser with WebGPU support.
Check it out! Link to repo blog post features and technical details in the reply. 🧵