Merge "[API] Support Multiple Attention Types" into main