Add cuDNN support for RNNs
Open
enhancement
help wanted
- Dominant language
- C++
- Stars
- 3.4k
- Forks
- 701
- PR merge metrics
- No merged PRs in 30d
Description
It would be nice to have cuDNN speedup for SimpleRNN, LSTM and GRU.
While cuDNN will likely not be able to accelerate incremental RNN construction, we can special case the `.transduce(seq)` and `.add_inputs(seq)` implementations to make use of the acceleration.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.