Release JetStream v0.2.2 (#110)

Author: JoeZijunZhou

README.md

@@ -29,7 +29,7 @@ Follow the steps in
 ## Get the jetstream-pytorch code
 ```bash
 git clone https://github.com/google/jetstream-pytorch.git
-git checkout v0.2.2
+git checkout jetstream-v0.2.2
 ```
 
 (optional) Create a virtual env using `venv` or `conda` and activate it.

pyproject.toml

@@ -3,7 +3,7 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [project]
-version = "0.2.1"
+version = "0.2.2"
 name = "jetstream_pt"
 dependencies = [
     "absl-py",