We explored different sequence models: Transformers, GRUs, LSTMs, S4d, S5.
To our surprise, we found GRUs to be far-and-away the most effective, and Transformers to be disappointingly ineffective.
Why? The combined F^T x B representation seems unstable for all non-GRU methods.