Index expression inference should match the results of pywave's index sequence analysis for vanilla attention