[TRAX] Bump version to 1.3.6

PiperOrigin-RevId: 338118536

Author: afrozenator

setup.py

@@ -21,7 +21,7 @@
 
 setup(
     name='trax',
-    version='1.3.5',
+    version='1.3.6',
     description='Trax',
     long_description=(
         'Trax helps you understand deep learning. We start with basic maths and'