What problem does it solve?
This Skill provides a set of commands to manage GKE TPU workloads, TPU nodepools, TPU topology, reservations, or Kubernetes Jobs on TPU v6e/v7x, streamlining the process of deploying and managing TPU resources.
Core Features & Use Cases
- GKE TPU Workload Management: Plan, render, and apply multi-doc Job manifests for GKE TPU workloads.
- Nodepool Management: Plan and create TPU nodepools with specific topology and machine types.
- Workload Rendering: Render one multi-doc Job manifest to a specified directory.
- Cleanup: Delete workload plans and nodepool plans for destructive actions.
- Use Case: If you need to deploy a TPU-based machine learning workload on GKE, this Skill can help you plan and create the necessary resources.
Quick Start
Run the plan-nodepool command with the appropriate configuration file to plan a TPU nodepool.