AI-Hypercomputer / jetstream-pytorch

PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
Apache License 2.0
42 stars 15 forks source link

Update Jetstream, add optional sampler args. #177

Closed qihqi closed 3 months ago