Add custom embedding

#53
by bcui19 - opened

Adds a custom embedding layer, which will allow us to work with FSDP more readily.

daking changed pull request status to merged

Sign up or log in to comment