GitHub repository: liurupeng/jetstream-pytorch
Description: PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"