Bump MaxText version to 0.2.0

Author: bvandermoon

src/MaxText/__init__.py

@@ -19,7 +19,7 @@
 """
 
 __author__ = "Google LLC"
-__version__ = "0.1.1"
+__version__ = "0.2.0"
 __description__ = (
     "MaxText is a high performance, highly scalable, open-source LLM written in pure Python/Jax and "
     "targeting Google Cloud TPUs and GPUs for training and **inference."