Alternatives and similar repositories for optimum-tpu:

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

• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆60Updated last month
• snowflakedb / ArcticTraining
  ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)
  ☆71Updated this week
• cognitivecomputations / spectrum
  ☆115Updated 3 weeks ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆115Updated last week
• huggingface / llm-swarm
  Manage scalable open LLM inference endpoints in Slurm clusters
  ☆254Updated 9 months ago
• euclaise / supertrainer2000
  ☆49Updated last year
• AI-Hypercomputer / maxdiffusion
  ☆201Updated this week
• huggingface / inference-benchmarker
  Inference server benchmarking tool
  ☆56Updated last week
• JoeLi12345 / nGPT
  an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)
  ☆98Updated last month
• VITA-Group / Q-GaLore
  Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.
  ☆198Updated 9 months ago
• Pleias / Various-Finetuning
  Set of scripts to finetune LLMs
  ☆37Updated last year
• HeegyuKim / torch-xla-SPMD
  Pytorch/XLA SPMD Test code in Google TPU
  ☆23Updated last year
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆105Updated this week
• cognitivecomputations / grokadamw
  ☆129Updated 8 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆244Updated this week
• huggingface / bloom-jax-inference
  ☆67Updated 2 years ago
• warner-benjamin / optimi
  Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
  ☆92Updated 9 months ago
• LLM360 / amber-data-prep
  Data preparation code for Amber 7B LLM
  ☆89Updated 11 months ago
• neuralmagic / nm-vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆262Updated 6 months ago
• AblateIt / finetune-study
  Comprehensive analysis of difference in performance of QLora, Lora, and Full Finetunes.
  ☆82Updated last year
• NVIDIA / NeMo-Run
  A tool to configure, launch and manage your machine learning experiments.
  ☆144Updated this week
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆218Updated last year
• cloneofsimo / min-fsdp
  ☆78Updated 10 months ago
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆85Updated last week
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆126Updated 5 months ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆82Updated last year
• NetEase-FuXi / EETQ
  Easy and Efficient Quantization for Transformers
  ☆197Updated 2 months ago
• epfml / DenseFormer
  ☆80Updated last year
• HanGuo97 / lq-lora
  ☆125Updated last year
• apple / ml-recurrent-drafter
  ☆209Updated 3 months ago