Fix TPU head resource name for v4 and v5e

AI-Hypercomputer / jetstream-pytorch

PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"

Apache License 2.0

41 stars 15 forks source link

Closed richardsliu closed 4 months ago