Skip to content

v0.0.6

Compare
Choose a tag to compare
@dianaml0 dianaml0 released this 24 Nov 16:18
· 931 commits to main since this release
2eb8b65

Fixed

  • Fix self attention optimization not being triggered, broken residual path [#119]
  • Improve speed by not using contiguous Tensors when not needed [#119]

Added

  • Attention mask wrapper [#113]
  • ViT comparison benchmark [#117]