How we made AWS Trainium 17x faster (for conv1d)
charleshong3.github.io