Update README (#30)

* Update README

* Update python pkg setup

Author: JoeZijunZhou

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
 

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",