[TRAX] Disable the following tests, they are more like integration tests and one of them actually required TPUs to run.
[TRAX] Disable the following tests, they are more like integration tests and one of them actually required TPUs to run.

```
trax/models/reformer/reformer_e2e_test.py
trax/models/reformer/reformer_memory_test.py
trax/models/reformer/reformer_oom_test.py
```

PiperOrigin-RevId: 318663926

google/trax,google/trax