[Misc] Update torchax with fp4 support (#1257)

Author: kyuyeunk

requirements.txt

@@ -9,7 +9,7 @@ jax[tpu]==0.8.0
 jaxlib==0.8.0
 jaxtyping
 flax==0.11.1
-torchax==0.0.7
+torchax==0.0.10
 qwix==0.1.1
 torchvision==0.24.0
 pathwaysutils