daking's picture bcui19's picture
Change `wte` to use shared embedding (#38)
dd6ba6a