Alternatives and similar repositories for tpu-training-example

Users that are interested in tpu-training-example are comparing it to the libraries listed below

• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆55Updated 3 months ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 7 months ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆48Updated 2 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated last year
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• google-deepmind / asyncdiloco
  ☆47Updated last year
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆18Updated 5 months ago
• EleutherAI / improved-t5
  Experiments for efforts to train a new and improved t5
  ☆76Updated last year
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆79Updated last year
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆85Updated 8 months ago
• siyan-zhao / prepacking
  The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …
  ☆60Updated last year
• RobertCsordas / moeut
  ☆91Updated last year
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆87Updated last year
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated last year
• graphcore-research / jax-scalify
  JAX Scalify: end-to-end scaled arithmetics
  ☆17Updated last year
• ethansmith2000 / TransformerExperiments
  ☆19Updated last month
• PythonNut / superbpe
  Official code release for "SuperBPE: Space Travel for Language Models"
  ☆77Updated last month
• Doraemonzzz / Awesome-Triton-Resources
  Awesome Triton Resources
  ☆39Updated 8 months ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 7 months ago
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆62Updated last year
• sustcsonglin / mamba-triton
  ☆51Updated last year
• UmerHA / triton_util
  Make triton easier
  ☆50Updated last year
• main-horse / hnet-old
  H-Net Dynamic Hierarchical Architecture
  ☆80Updated 3 months ago
• cloneofsimo / min-fsdp
  ☆92Updated last year
• kyleliang919 / Super_Muon
  ☆66Updated 9 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆131Updated last year
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆23Updated 7 months ago
• yifanzhang-pro / HLA
  Official Project Page for HLA: Higher-order Linear Attention (https://arxiv.org/abs/2510.27258)
  ☆42Updated 2 weeks ago