athms / mad-lab

Related projects: ⓘ

• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆156Updated 4 months ago
• berlino / seq_icl
  ☆48Updated 3 months ago
• shikaiqiu / compute-better-spent
  ☆42Updated 3 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 8 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆94Updated 2 weeks ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆129Updated 2 months ago
• epfml / schedules-and-scaling
  ☆47Updated 3 months ago
• epfml / llm-baselines
  ☆68Updated 3 weeks ago
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆76Updated 2 years ago
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆206Updated last month
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆108Updated 3 weeks ago
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆91Updated 5 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆74Updated 7 months ago
• epfml / dynamic-sparse-flash-attention
  ☆129Updated last year
• taufeeque9 / codebook-features
  Sparse and discrete interpretability tool for neural networks
  ☆51Updated 7 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆66Updated 7 months ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆86Updated 3 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆110Updated 5 months ago
• kuleshov-group / mdlm
  Simplified Masked Diffusion Language Model
  ☆160Updated last week
• srush / do-we-need-attention
  ☆160Updated last year
• epfml / DenseFormer
  ☆73Updated 5 months ago
• cloneofsimo / min-fsdp
  ☆68Updated 2 months ago
• EleutherAI / improved-t5
  Experiments for efforts to train a new and improved t5
  ☆76Updated 5 months ago
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆60Updated this week
• lucidrains / pause-transformer
  Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…
  ☆50Updated 10 months ago
• google-deepmind / nanodo
  ☆180Updated 2 months ago
• insuhan / hyper-attn
  ☆65Updated 9 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆116Updated last month
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆57Updated 10 months ago
• srush / mamba-primer
  ☆35Updated 5 months ago