xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.
☆193Aug 27, 2026Updated 3 weeks ago
Alternatives and similar repositories for xpk
Users that are interested in xpk are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Package of Pathways-on-Cloud utilities☆33Updated this week
- ☆127Sep 5, 2026Updated 2 weeks ago
- Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.☆140Sep 8, 2026Updated last week
- ☆386Updated this week
- ☆18Apr 8, 2026Updated 5 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- PathwaysJob API is an OSS Kubernetes-native API, to deploy ML training and batch inference workloads, using Pathways on GKE.☆23Jul 9, 2026Updated 2 months ago
- KJob: Tool for CLI-loving ML researchers☆44Jun 1, 2026Updated 3 months ago
- A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across differe…☆64Updated this week
- ☆17Jan 23, 2026Updated 7 months ago
- torchprime is a reference model implementation for PyTorch on TPU.☆49Mar 3, 2026Updated 6 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…☆458Jan 5, 2026Updated 8 months ago
- A simple, performant, and scalable Jax LLM!☆2,421Updated this week
- ☆15May 11, 2025Updated last year
- ☆38Oct 31, 2025Updated 10 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Tokamax: A GPU and TPU kernel library.☆294Updated this week
- ☆22Mar 11, 2026Updated 6 months ago
- a Jax quantization library☆134Updated this week
- ☆26Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆535Updated this week
- Minimal yet performant LLM examples in pure JAX☆281Aug 23, 2026Updated 3 weeks ago
- ☆156Updated this week
- ☆32Sep 9, 2026Updated last week
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆85Dec 18, 2025Updated 9 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆592Jul 11, 2024Updated 2 years ago
- TPU inference for vLLM, with unified JAX and PyTorch support.☆435Updated this week
- JAX-Toolbox☆433Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆560Updated this week
- Cluster Toolkit is an open-source software offered by Google Cloud which makes it easy for customers to deploy AI/ML and HPC environments…☆361Updated this week
- Stochastic trace estimation using JAX☆22Aug 27, 2026Updated 3 weeks ago
- JobSet: a k8s native API for distributed ML training and HPC workloads☆344Updated this week
- This repository is a collection of accelerated platform best practices, reference architectures, example use cases, reference implementat…☆103Updated this week
- AI on GKE is a collection of examples, best-practices, and prebuilt solutions to help build, deploy, and scale AI Platforms on Google Kub…☆330Jun 23, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ☆19Feb 18, 2026Updated 7 months ago
- ☆226Updated this week
- Library for reading and processing ML training data.☆781Updated this week
- ☆362Updated this week
- JAX backend for SGL☆352Updated this week
- A JAX-native High Performance Eval Metrics Library☆64Updated this week
- ☆312Aug 31, 2026Updated 2 weeks ago