Alternatives and similar repositories for lit-llama

Users that are interested in lit-llama are comparing it to the libraries listed below

• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• cat-state / tinypar
  ☆20Updated 2 years ago
• ethansmith2000 / TransformerExperiments
  ☆19Updated 3 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• haileyschoelkopf / triton-index
  See https://github.com/cuda-mode/triton-index/ instead!
  ☆11Updated last year
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 3 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• EleutherAI / training-jacobian
  ☆23Updated 8 months ago
• berlino / seq_icl
  ☆53Updated last year
• 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…
  ☆53Updated last year
• cloneofsimo / min-fsdp
  ☆87Updated last year
• sholtodouglas / scalingExperiments
  ☆61Updated 3 years ago
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆69Updated last year
• BlinkDL / SmallInitEmb
  LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence
  ☆59Updated 3 years ago
• samblouir / birdie
  ☆13Updated 3 months ago
• main-horse / hnet
  H-Net Dynamic Hierarchical Architecture
  ☆79Updated last month
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• Jaykef / Triton-nanoGPT
  Custom triton kernels for training Karpathy's nanoGPT.
  ☆19Updated 10 months ago
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆23Updated 3 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated 8 months ago
• cloneofsimo / zeroshampoo
  ☆34Updated 11 months ago
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆82Updated 3 years ago
• JonasGeiping / linear_cross_entropy_loss
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆70Updated last year
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated last year
• codekansas / rwkv
  RWKV model implementation
  ☆38Updated 2 years ago
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆54Updated last year
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆48Updated last year
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆60Updated last week
• lianakoleva / no-libtorch-compile
  ☆21Updated 6 months ago