The Local LLM Index / Inference Engines / #208

AI-Hypercomputer/JetStream

by AI-Hypercomputer · Inference Engines · updated 9mo ago

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

30
momentum
472
stars
70
forks
#208
rank
gemmagptgpuinferencejaxlarge-language-modelsllamallama2llmllm-inferencellmopsmlops
View on GitHub →