Update version to 0.0.6 (#9611)

Author: qihqi

torchax/torchax/__init__.py

@@ -8,7 +8,7 @@
 from torchax import tensor
 from contextlib import contextmanager
 
-__version__ = "0.0.5"
+__version__ = "0.0.6"
 VERSION = __version__
 
 __all__ = [