Remove reference to GPU, its not supported yet. We can add back when the support is ready.

Author: vipannalla

README.md

@@ -1,8 +1,8 @@
-# JetStream - A throughput and memory optimized engine for LLM inference on TPU and GPU
+# JetStream - A throughput and memory optimized engine for LLM inference on TPUs
 
 ## About
 
-JetStream is a fast library for LLM inference and serving on TPU and GPU.
+JetStream is a fast library for LLM inference and serving on TPUs.
 
 ## Getting Started
 

setup.py

@@ -26,8 +26,7 @@ def parse_requirements(filename):
     name='google-jetstream',
     version='0.1.1',
     description=(
-        'A throughput and memory optimized engine for LLM inference on TPU and'
-        ' GPU.'
+        'A throughput and memory optimized engine for LLM inference on TPUs'
     ),
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',