Alternatives and similar repositories for train-tk:

Users that are interested in train-tk are comparing it to the libraries listed below

• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆68Updated 11 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆117Updated 3 months ago
• UmerHA / triton_util
  Make triton easier
  ☆45Updated 8 months ago
• cloneofsimo / min-fsdp
  ☆75Updated 7 months ago
• PiotrNawrot / nano-sparse-attention
  The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.
  ☆57Updated last month
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆105Updated 2 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆76Updated last year
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆81Updated 2 weeks ago
• siyan-zhao / prepacking
  The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"
  ☆59Updated 4 months ago
• EleutherAI / improved-t5
  Experiments for efforts to train a new and improved t5
  ☆77Updated 10 months ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆99Updated 3 months ago
• kyleliang919 / Online-Subspace-Descent
  This repo is based on https://github.com/jiaweizzhao/GaLore
  ☆25Updated 5 months ago
• berlino / seq_icl
  ☆51Updated 9 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 4 months ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆47Updated last week
• cat-state / tinypar
  ☆20Updated last year
• AnswerDotAI / cold-compress
  Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…
  ☆119Updated 6 months ago
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆186Updated 9 months ago
• minyoungg / LTE
  ☆67Updated 7 months ago
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆17Updated 3 weeks ago
• google-deepmind / asyncdiloco
  ☆42Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 2 months ago
• srush / LLM-Talk
  ☆48Updated last year
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆96Updated 11 months ago
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆79Updated 2 years ago
• abacusai / gh200-llm
  Docker image NVIDIA GH200 machines - optimized for vllm serving and hf trainer finetuning
  ☆36Updated last week
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆222Updated 2 weeks ago
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆40Updated 5 months ago