Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement RNN/LSTMs for Leaf #73

Open
MichaelHirn opened this issue Mar 8, 2016 · 5 comments
Open

Implement RNN/LSTMs for Leaf #73

MichaelHirn opened this issue Mar 8, 2016 · 5 comments

Comments

@MichaelHirn
Copy link
Member

This issue makes part of #20 more concrete.

Recurrent Neural Networks, became an effective Neural Network architecture, that we would like to implement in Leaf as well. The operations could probably be composed of BLAS operations with collenchyma-blas. Differentiation for backpropagation is another topic, that might better be solved with Auto Differentiation.

@johnnyman727
Copy link

@MichaelHirn if this isn't super timer critical, I can own this task. I will need some hand-holding to get started. Where would be the best place to start?

@MichaelHirn
Copy link
Member Author

Awesome!

I think it should be verily straight forward. Two resources about RNN and performance, that look interesting and might be helpful for technical implementation.

The rest is just adding a new common layer to Leaf, using the BLAS operations from collenchyma-nn. You can reach @hobofan anytime on the gitter chat for any guidance.

If you succeed we the implementation, we are super thrilled to publish the Benchmarks of your RNN/LSTM implementation on Deep Learning Benchmarks and compare them with RNN implementations of other Frameworks.

@MichaelHirn
Copy link
Member Author

As an update:

cuDNN v5, which will be released for early developers in a few weeks, will come with primitives for LSTM/RNNs.

@carlthome
Copy link

Adding most of the optimizations described in cuDNN 5 here would easily make leaf the fastest basic LSTM implementation out there.

@robsmith11
Copy link

Was any progress made on this? I might try to add LSTM support in the future.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

4 participants