search

probml / dynamax

State Space Models library in JAX
https://probml.github.io/dynamax/
MIT License
667 stars 76 forks source link

make nonlinear SSM for text generation #280

Closed murphyk closed 1 year ago

murphyk commented 1 year ago

See eg F. Schmidt and T. Hofmann, “Deep state space models for unconditional word generation,” arXiv [cs.LG], Jun. 12, 2018 [Online]. Available: https://proceedings.neurips.cc/paper/2018/file/0cd60efb5578cd967c3c23894f305800-Paper.pdf. [Accessed: Nov. 14, 2022] See also S5.