Update README (#30)

* Update README * Update python pkg setup
AI-Hypercomputer · Apr 4, 2024 · 10bd138 · 10bd138
1 parent 385ecac
commit 10bd138
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,8 +1,8 @@
-# JetStream - A throughput and memory optimized engine for LLM inference on TPUs
+# JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
 
 ## About
 
-JetStream is a fast library for LLM inference and serving on TPUs.
+JetStream is a fast library for LLM inference and serving on TPUs (and GPUs in future -- PRs welcome).
 
 ## Getting Started
 

diff --git a/setup.py b/setup.py
@@ -26,7 +26,7 @@ def parse_requirements(filename):
     name="google-jetstream",
     version="0.1.1",
     description=(
-        "A throughput and memory optimized engine for LLM inference on TPUs"
+        "JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome)."
     ),
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",