Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
AlphaUncommon

AI-Hypercomputer/JetStream

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

Owner
AI-Hypercomputer (organization)
Stars
Stars: 472
Forks
Forks: 70
Language
Python
License
Apache-2.0
Created
2024
Topics
gemmagptgpuinferencejaxlarge-language-modelsllamallama2llmmodel-serving
Tier
Alpha (250 to 699 ★) What this tier means
View on GitHub

More repos by AI-Hypercomputer

All 8 repos by AI-Hypercomputer

Nearby in Python

All Python repos