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.
☆169Feb 20, 2026Updated last week
Alternatives and similar repositories for xpk
Users that are interested in xpk are comparing it to the libraries listed below
Sorting:
- ☆77Updated this week
- Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.☆117Updated 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…☆413Jan 5, 2026Updated last month
- Package of Pathways-on-Cloud utilities☆25Updated this week
- ☆14Feb 18, 2025Updated last year
- A simple, performant and scalable Jax LLM!☆2,148Updated this week
- ☆151Feb 12, 2026Updated 2 weeks ago
- 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
- ☆192Feb 16, 2026Updated last week
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Jun 11, 2023Updated 2 years ago
- ☆566Jul 11, 2024Updated last year
- Orbax provides common checkpointing and persistence utilities for JAX users☆482Updated this week
- ☆348Feb 19, 2026Updated last 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…☆327Jun 23, 2025Updated 8 months ago
- Google TPU optimizations for transformers models☆134Jan 23, 2026Updated last month
- Library for reading and processing ML training data.☆685Updated this week
- Task-based datasets, preprocessing, and evaluation for sequence models.☆594Feb 3, 2026Updated 3 weeks ago
- JAX-Toolbox☆385Updated this week
- ☆15May 11, 2025Updated 9 months ago
- 🚀 Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.☆13Jan 30, 2026Updated last month
- Collection of tools and examples for managing Accelerated workloads in Kubernetes Engine☆248Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆439Updated 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…☆322Updated this week
- Notebooks for managing NeurIPS 2014 and analysing the NeurIPS experiment.☆13May 22, 2024Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Tokamax: A GPU and TPU kernel library.☆176Feb 21, 2026Updated last week
- ☆61Updated this week
- ☆79Updated this week
- Holistic job manager on Kubernetes☆116Feb 20, 2024Updated 2 years ago
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- ☆20Apr 24, 2025Updated 10 months ago
- 5th Place Solution to 3rd YouTube-8M Video Understanding Challenge (Last Top GB Model)☆13Oct 23, 2019Updated 6 years ago
- PathwaysJob API is an OSS Kubernetes-native API, to deploy ML training and batch inference workloads, using Pathways on GKE.☆19Oct 22, 2025Updated 4 months ago
- An Extensible Deep Learning Library☆2,318Feb 18, 2026Updated last week
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- Demonstrates iterative FGSM on Apple's NeuralHash model.☆16Aug 19, 2021Updated 4 years ago
- ☆16Oct 20, 2025Updated 4 months ago
- Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers☆27Mar 1, 2025Updated 11 months ago
- Kubernetes-native Job Queueing☆2,329Updated this week