issues
search
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
Changes
Embed
layerEmbed