Stars
A simple, performant and scalable Jax LLM!
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
An Open Source Machine Learning Framework for Everyone
{{ message }}