clab/dynet

Add cuDNN support for RNNs

Open

#574 ouverte le 31 mai 2017

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)C++ (701 forks)github user discovery
enhancementhelp wanted

Métriques du dépôt

Stars
 (3 433 stars)
Métriques de merge PR
 (Métriques PR en attente)

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.

Guide contributeur

Direction de recherche
Étudiez l'API RNN de cuDNN et intégrez la dans les implémentations SimpleRNN, LSTM et GRU de DyNet, en vous concentrant sur les méthodes .transduce() et .add inputs(). Assurez la compatibilité avec les abstractions RNN existantes de DyNet.
Stack technique
cpp
Domaine
machine learning
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
4
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Récente
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
C++CUDAcuDNNDyNet internals
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
20