[assume_pure] test_assume_pure_complex_module torchax output on TPU differs from torch_xla #8974
Description
To repro, re-enable the test and run python3 test/test_assume_pure.py
{{ message }}
To repro, re-enable the test and run python3 test/test_assume_pure.py