Pathways-on-Cloud Utilities

Pathwaysutils is a Python package providing essential utilities and tools to streamline the deployment and execution of JAX workloads on the Pathways on Cloud architecture. It simplifies large-scale machine learning computations across multiple TPU slices by handling cloud-specific adaptations. Pathways is used internally at Google for models like Gemini, and this package brings similar benefits to Google Cloud customers. The current version is 0.1.7, and it is actively developed with releases tightly coupled to JAX versions.