xjdr-alt / simple_transformerLinks
Simple Transformer in Jax
☆137Updated 11 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:
- smol models are fun too☆92Updated 6 months ago
- smolLM with Entropix sampler on pytorch☆150Updated 7 months ago
- SIMD quantization kernels☆65Updated last week
- Just large language models. Hackable, with as little abstraction as possible. Done for my own purposes, feel free to rip.☆44Updated last year
- Compiling useful links, papers, benchmarks, ideas, etc.☆46Updated 2 months ago
- look how they massacred my boy☆63Updated 7 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆100Updated 2 months ago
- A graph visualization of attention☆55Updated 2 weeks ago
- Plotting (entropy, varentropy) for small LMs☆96Updated 2 weeks ago
- ☆111Updated 5 months ago
- supporting pytorch FSDP for optimizers☆79Updated 5 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆76Updated last month
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆64Updated 7 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆66Updated last month
- ☆267Updated 10 months ago
- The history files when recording human interaction while solving ARC tasks☆110Updated last week
- Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆173Updated 10 months ago
- An introduction to LLM Sampling☆78Updated 5 months ago
- in this repository, i'm going to implement increasingly complex llm inference optimizations☆58Updated last week
- ☆27Updated 10 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆126Updated 3 weeks ago
- seqax = sequence modeling + JAX☆155Updated last month
- A really tiny autograd engine☆94Updated last week
- Comprehensive analysis of difference in performance of QLora, Lora, and Full Finetunes.☆81Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆116Updated 5 months ago
- Ultra low overhead NVIDIA GPU telemetry plugin for telegraf with memory temperature readings.☆62Updated 10 months ago
- MiniHF is an inference, human preference data collection, and fine-tuning tool for local language models. It is intended to help the user…☆172Updated last week
- This repository contain the simple llama3 implementation in pure jax.☆64Updated 3 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆50Updated 6 months ago
- DeMo: Decoupled Momentum Optimization☆188Updated 6 months ago