poets-ai / elegy

A High Level API for Deep Learning in JAX
https://poets-ai.github.io/elegy/
MIT License
469 stars 32 forks source link

LSTM #242

Open mwitiderrick opened 2 years ago

mwitiderrick commented 2 years ago

Is there a Flax LSTM example?