Commit Graph

7 Commits (9d9959919ba2d615f4d37a11e2abb317b5a62357)

Author SHA1 Message Date
Ross Wightman ab49d275de Significant norm update
4 years ago
Alexander Soare b25ff96768 wip - pre-rebase
4 years ago
Alexander Soare e051dce354 Make all models FX traceable
4 years ago
Alexander Soare 0149ec30d7 wip - attempting to rebase
4 years ago
Alexander Soare bc3d4eb403 wip -rebase
4 years ago
Ross Wightman b7a568f065 Fix torchscript issue in bat
4 years ago
Ross Wightman 307a935b79 Add non-local and BAT attention. Merge attn and self-attn factories into one. Add attention references to README. Add mlp 'mode' to ECA.
4 years ago