TPU inference for vLLM, with unified JAX and PyTorch support.
☆247Mar 4, 2026Updated this week
Alternatives and similar repositories for tpu-inference
Users that are interested in tpu-inference are comparing it to the libraries listed below
Sorting:
- vLLM performance dashboard☆42Apr 26, 2024Updated last year
- ☆16Feb 18, 2026Updated 2 weeks ago
- ☆314Feb 27, 2026Updated last week
- ☆13Updated this week
- ☆18Jun 18, 2025Updated 8 months ago
- ☆15May 11, 2025Updated 9 months ago
- Kernel Library Wheel for SGLang☆16Updated this week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆414Jan 5, 2026Updated 2 months ago
- (WIP) Parallel inference for black-forest-labs' FLUX model.☆19Nov 18, 2024Updated last year
- JAX backend for SGL☆243Updated this week
- Flexibly track outputs and grad-outputs of torch.nn.Module.☆13Oct 6, 2023Updated 2 years ago
- Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.☆118Updated this week
- Large Language Model Text Generation Inference on Habana Gaudi☆34Mar 20, 2025Updated 11 months ago
- Paper-reading notes for Berkeley OS prelim exam.☆14Aug 28, 2024Updated last year
- Benchmark tests supporting the TiledCUDA library.☆18Nov 19, 2024Updated last year
- JAX bindings for the flash-attention3 kernels☆21Jan 2, 2026Updated 2 months ago
- Triton kernels for Flux☆22Jul 7, 2025Updated 7 months ago
- A Top-Down Profiler for GPU Applications☆22Feb 29, 2024Updated 2 years ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆219Updated this week
- ☆135Dec 22, 2025Updated 2 months ago
- Multi-Turn RL Training System with AgentTrainer for Language Model Game Reinforcement Learning☆60Dec 18, 2025Updated 2 months ago
- Google TPU optimizations for transformers models☆134Jan 23, 2026Updated last month
- Lightweight Python Wrapper for OpenVINO, enabling LLM inference on NPUs☆27Dec 17, 2024Updated last year
- A high-throughput and memory-efficient inference and serving engine for LLMs☆85Updated this week
- Quantized Attention on GPU☆44Nov 22, 2024Updated last year
- ☆20Nov 23, 2022Updated 3 years ago
- Rust crate for some audio utilities☆27Mar 8, 2025Updated 11 months ago
- Website with current metrics on the fastest AI models.☆43Nov 13, 2024Updated last year
- A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across differe…☆58Updated this week
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆143Dec 4, 2024Updated last year
- ☆193Feb 27, 2026Updated last week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆25Nov 7, 2025Updated 4 months ago
- Automatic differentiation for Triton Kernels☆29Aug 12, 2025Updated 6 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆93Sep 4, 2024Updated last year
- ☆77Feb 23, 2026Updated last week
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆716Feb 28, 2026Updated last week
- Genai-bench is a powerful benchmark tool designed for comprehensive token-level performance evaluation of large language model (LLM) serv…☆275Updated this week
- ☆566Jul 11, 2024Updated last year
- Triton-based implementation of Sparse Mixture of Experts.☆268Oct 3, 2025Updated 5 months ago