huggingface / optimum-tpuLinks
Google TPU optimizations for transformers models
☆113Updated 5 months ago
Alternatives and similar repositories for optimum-tpu
Users that are interested in optimum-tpu are comparing it to the libraries listed below
Sorting:
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆61Updated 2 months ago
- Load compute kernels from the Hub☆172Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆264Updated 8 months ago
- ☆212Updated 4 months ago
- ☆222Updated this week
- Repo hosting codes and materials related to speeding LLMs' inference using token merging.☆36Updated last year
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆198Updated 11 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆253Updated this week
- ☆65Updated last week
- torchprime is a reference model implementation for PyTorch on TPU.☆28Updated this week
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆188Updated this week
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆119Updated this week
- vLLM: A high-throughput and memory-efficient inference and serving engine for LLMs☆87Updated last week
- Lightweight toolkit package to train and fine-tune 1.58bit Language models☆78Updated last month
- A safetensors extension to efficiently store sparse quantized tensors on disk☆126Updated this week
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆277Updated last year
- Inference server benchmarking tool☆73Updated last month
- Collection of autoregressive model implementation☆85Updated last month
- A tool to configure, launch and manage your machine learning experiments.☆161Updated this week
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆101Updated 3 months ago
- Pytorch/XLA SPMD Test code in Google TPU☆23Updated last year
- PyTorch per step fault tolerance (actively under development)☆329Updated this week
- Manage scalable open LLM inference endpoints in Slurm clusters☆260Updated 11 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 10 months ago
- Easy and Efficient Quantization for Transformers☆199Updated 4 months ago
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆116Updated 6 months ago
- Fully fine-tune large models like Mistral, Llama-2-13B, or Qwen-14B completely for free☆231Updated 7 months ago
- SGLang is fast serving framework for large language models and vision language models.☆23Updated 4 months ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆349Updated last week
- ☆132Updated 10 months ago