HeegyuKim / torch-xla-SPMD

Related projects ⓘ

Alternatives and complementary repositories for torch-xla-SPMD

• bminixhofer / zett
  Code for Zero-Shot Tokenizer Transfer
  ☆117Updated last month
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆71Updated last month
• ChrisHayduk / qlora-multi-gpu
  QLoRA with Enhanced Multi GPU Support
  ☆36Updated last year
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆67Updated this week
• cloneofsimo / fim-llama-deepspeed
  ☆31Updated 10 months ago
• melisa-writer / short-transformers
  Prune transformer layers
  ☆65Updated 5 months ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆84Updated this week
• sanyalsunny111 / LLM-Inheritune
  This is the official repository for Inheritune.
  ☆105Updated last month
• cloneofsimo / min-fsdp
  ☆73Updated 4 months ago
• mgmalek / efficient_cross_entropy
  ☆77Updated 5 months ago
• euclaise / supertrainer2000
  ☆49Updated 8 months ago
• kyleliang919 / Long-context-transformers
  Exploring finetuning public checkpoints on filter 8K sequences on Pile
  ☆115Updated last year
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆94Updated 7 months ago
• TRI-ML / linear_open_lm
  A repository for research on medium sized language models.
  ☆74Updated 6 months ago
• kyleliang919 / Online-Subspace-Descent
  This repo is based on https://github.com/jiaweizzhao/GaLore
  ☆19Updated 2 months ago
• akjindal53244 / Arithmo
  Small and Efficient Mathematical Reasoning LLMs
  ☆71Updated 9 months ago
• Zyphra / Zyda_processing
  ☆28Updated 5 months ago
• epfml / DenseFormer
  ☆77Updated 7 months ago
• r-three / RAD
  Reference implementation for Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model
  ☆41Updated 10 months ago
• lucidrains / llama-qrlhf
  Implementation of the Llama architecture with RLHF + Q-learning
  ☆157Updated 11 months ago
• snu-mllab / Context-Memory
  Pytorch implementation for "Compressed Context Memory For Online Language Model Interaction" (ICLR'24)
  ☆50Updated 7 months ago
• cat-state / tinypar
  ☆20Updated last year
• minyoungg / LTE
  ☆64Updated 4 months ago
• kernelmachine / cbtm
  Code repository for the c-BTM paper
  ☆105Updated last year
• erfanzar / jax-flash-attn2
  Flash Attention Implementation with Multiple Backend Support and Sharding This module provides a flexible implementation of Flash Attenti…
  ☆18Updated last week
• epfml / schedules-and-scaling
  ☆53Updated 3 weeks ago
• ml-jku / EVA
  One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation
  ☆30Updated last month
• 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…
  ☆49Updated last year
• ContextualAI / CLAIR_and_APO
  Anchored Preference Optimization and Contrastive Revisions: Addressing Underspecification in Alignment
  ☆46Updated 2 months ago
• argilla-io / distilabel-spin-dibt
  Repository containing the SPIN experiments on the DIBT 10k ranked prompts
  ☆23Updated 8 months ago