Update tree_math version number

PiperOrigin-RevId: 533175662

Author: shoyer

setup.py

@@ -29,7 +29,7 @@
 setuptools.setup(
     name='tree-math',
     description='Mathematical operations for JAX pytrees',
-    version='0.1.0 ',
+    version='0.2.0 ',
     license='Apache 2.0',
     author='Google LLC',
     author_email='[email protected]',