Commit ffd6a30
Switch WMT default attention to JAX built-in attention, switch default launch script to be on one cell for both compute & storage
PiperOrigin-RevId: 8592775511 parent bc3c6d9 commit ffd6a30
2 files changed
Lines changed: 366 additions & 241 deletions
0 commit comments