Adds a custom embedding layer, which will allow us to work with FSDP more readily.
· Sign up or log in to comment
