Initial commit
Merge https://github.com/kevinzenghu/attention-bandwidth