Chillee / lit-llama

Related projects ⓘ

Alternatives and complementary repositories for lit-llama

• ethansmith2000 / TransformerExperiments
  ☆18Updated last month
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 3 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆32Updated last year
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago
• Jaykef / Triton-nanoGPT
  ☆13Updated last month
• cat-state / tinypar
  ☆20Updated last year
• lianakoleva / no-libtorch-compile
  ☆17Updated 3 weeks ago
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆67Updated last year
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆30Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆42Updated 5 months ago
• koayon / atp_star
  PyTorch and NNsight implementation of AtP* (Kramar et al 2024, DeepMind)
  ☆14Updated 7 months ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆46Updated 11 months ago
• ClashLuke / SOAP
  ☆19Updated last week
• haileyschoelkopf / triton-index
  See https://github.com/cuda-mode/triton-index/ instead!
  ☆11Updated 6 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated 2 weeks ago
• ckkissane / crosscoder-model-diff-replication
  Open source replication of Anthropic's Crosscoders for Model Diffing
  ☆16Updated 3 weeks ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated last year
• renll / SeqBoat
  [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling
  ☆35Updated 11 months ago
• HomebrewNLP / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆46Updated 10 months ago
• yixiaoer / tpu-training-example
  ☆13Updated 4 months ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆16Updated this week
• cloneofsimo / min-fsdp
  ☆73Updated 4 months ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 5 months ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆33Updated 4 months ago
• cloneofsimo / zeroshampoo
  ☆31Updated 2 months ago
• mgmalek / efficient_cross_entropy
  ☆77Updated 5 months ago
• berlino / seq_icl
  ☆50Updated 6 months ago