An Extensible Deep Learning Library

Package of Pathways-on-Cloud utilities

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.

A simple, performant and scalable Jax LLM!