JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
(451 stars) (67 forks) (0 件の索引済み issue) (0 件のオープンな good first issue)
リポジトリ
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
A simple, performant, and scalable Jax LLM!