cgarciae / treex

A Pytree Module system for Deep Learning in JAX

https://cgarciae.github.io/treex/

MIT License

215 stars 17 forks source link

Add Embed layer #54

Closed cgarciae closed 2 years ago

cgarciae commented 2 years ago

Changes

[x] Adds Embed layer
[x] Add tests for Embed