ethansmith2000 / TransformerExperimentsLinks
☆19Updated last month
Alternatives and similar repositories for TransformerExperiments
Users that are interested in TransformerExperiments are comparing it to the libraries listed below
Sorting:
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- ☆34Updated last year
- ☆35Updated last year
- Code for the paper "Function-Space Learning Rates"☆23Updated 7 months ago
- ☆82Updated last year
- ☆20Updated 2 years ago
- ☆53Updated 2 years ago
- Combining SOAP and MUON☆17Updated 11 months ago
- ☆92Updated last year
- research impl of Native Sparse Attention (2502.11089)☆63Updated 10 months ago
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆61Updated 3 years ago
- Automatically take good care of your preemptible TPUs☆37Updated 2 years ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated last year
- supporting pytorch FSDP for optimizers☆84Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Updated 5 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- Latent Diffusion Language Models☆70Updated 2 years ago
- Griffin MQA + Hawk Linear RNN Hybrid☆88Updated last year
- ☆50Updated last year
- Triton Implementation of HyperAttention Algorithm☆48Updated 2 years ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Updated 10 months ago
- Experiments on the impact of depth in transformers and SSMs.☆40Updated 2 months ago
- ☆70Updated last year
- ☆62Updated last year
- Experiment of using Tangent to autodiff triton☆81Updated last year
- ☆22Updated last year
- Efficient PScan implementation in PyTorch☆17Updated 2 years ago
- Focused on fast experimentation and simplicity☆79Updated last year
- GoldFinch and other hybrid transformer components☆45Updated last year
- Parallel Associative Scan for Language Models☆18Updated 2 years ago