PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
☆82Dec 18, 2025Updated 5 months ago
Alternatives and similar repositories for jetstream-pytorch
Users that are interested in jetstream-pytorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆437Jan 5, 2026Updated 4 months ago
- Google TPU optimizations for transformers models☆138Jan 23, 2026Updated 3 months ago
- Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)☆64May 5, 2026Updated 2 weeks ago
- ☆21Apr 27, 2026Updated 3 weeks ago
- ☆18Feb 18, 2026Updated 3 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- a Jax quantization library☆114Updated this week
- This repository contains example code to build models on TPUs☆30Feb 17, 2023Updated 3 years ago
- Latent Large Language Models☆19Aug 24, 2024Updated last year
- MLIR-based partitioning system☆184Updated this week
- A simple, performant and scalable Jax LLM!☆2,280Updated this week
- A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across differe…☆63Updated this week
- KernelBench v2: Can LLMs Write GPU Kernels? - Benchmark with Torch -> Triton (and more!) problems☆23Jul 4, 2025Updated 10 months ago
- sitp: run nanochat by building teenygrad from scratch: the bridge from micrograd to tinygrad!☆72Updated this week
- torchprime is a reference model implementation for PyTorch on TPU.☆47Mar 3, 2026Updated 2 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- PyTorch distributed training acceleration framework☆55Aug 13, 2025Updated 9 months ago
- ☆29Jun 3, 2024Updated last year
- JAX Implementations of Descript Audio Codec and EnCodec☆36Mar 30, 2025Updated last year
- Implementation of Direct Preference Optimization☆17Jul 17, 2023Updated 2 years ago
- JAX backend for SGL☆270Updated this week
- Minimal yet performant LLM examples in pure JAX☆255Apr 10, 2026Updated last month
- ☆30May 5, 2026Updated 2 weeks ago
- An IR for efficiently simulating distributed ML computation.☆33Jan 13, 2024Updated 2 years ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆705Jan 26, 2026Updated 3 months ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆181Updated this week
- ☆14Nov 28, 2022Updated 3 years ago
- Getting confidences from any end-to-end systems☆11May 24, 2023Updated 2 years ago
- FP4 MAC Array☆19Apr 14, 2024Updated 2 years ago
- ☆151May 6, 2026Updated 2 weeks ago
- ☆20Apr 24, 2026Updated 3 weeks ago
- Official repository for "BLEUBERI: BLEU is a surprisingly effective reward for instruction following"☆32Jun 5, 2025Updated 11 months ago
- See details in https://github.com/pytorch/xla/blob/r1.12/torch_xla/distributed/fsdp/README.md☆25Dec 22, 2022Updated 3 years ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆289May 13, 2026Updated last week
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- 한국어 언어 모델 학습을 위한 프로젝트(Flax, Pytorch with Huggingface Accelerate)☆31Sep 13, 2023Updated 2 years ago
- PyTorch centric eager mode debugger☆48Dec 16, 2024Updated last year
- Lattice combination algorithm to combine inaccurate transcripts with hypothesis lattices☆16Mar 19, 2024Updated 2 years ago
- Java tool to translate VRP instances to VRP-REP unified format.☆11Nov 28, 2014Updated 11 years ago
- Starter template for your ML/AI projects (uv package manager, RestAPI with FastAPI and Dockerfile support)☆35Jan 13, 2025Updated last year
- ☆34May 14, 2025Updated last year
- This repository sets up a sample microservice architecture to highlight how Cloud Spanner integrates with other cloud technologies.☆20Mar 18, 2026Updated 2 months ago