Commit Graph

7 Commits (c42e26f866d0adccbf1a74b7b5e1108ae6be13a2)

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