Megatron is a package/module to supplement keras by adding in Transformer and multiheaded attention mechanisms;
Note Megatron is likely to be deprecated if keras/tf decides to add a native attention mechanism.
-
Notifications
You must be signed in to change notification settings - Fork 1
Snarik/Megatron
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Keras implementations of multi-headed attention mechanisms
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published