Fix TPU head resource name for v4 and v5e

google / jetstream-pytorch

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

Apache License 2.0

33 stars 14 forks source link

Closed richardsliu closed 1 month ago