Skip to content
This repository was archived by the owner on Dec 15, 2025. It is now read-only.

Add in a GNMT implementation#16

Open
slyubomirsky wants to merge 2 commits intouwsampl:3la-rebase-completefrom
slyubomirsky:add-gnmt
Open

Add in a GNMT implementation#16
slyubomirsky wants to merge 2 commits intouwsampl:3la-rebase-completefrom
slyubomirsky:add-gnmt

Conversation

@slyubomirsky
Copy link
Collaborator

This PR adds in a Relay implementation of GNMT (the script tests that it runs), including Luong general attention, which is tested against a PyTorch reference implementation.

Please review @AD1024, @ttambe, and @LeeOHzzZ -- Mike, I think you should "take over" this PR to add in appropriate pattern matches

@slyubomirsky
Copy link
Collaborator Author

Note: This reuses code from the speech-to-text example, so we should probably deduplicate that by putting the LSTM definitions in a separate file

Co-authored-by: Mike He <dh63@cs.washington.edu>
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants