Credit to https://github.com/google-research/language/tree/master/language/asqa