/ntmcell_pytorch

PyTorch implementation of Neural Turing Machine recurrent neural network

Primary LanguagePython

PyTorch Neural Turing Machine (NTM)

PyTorch implementation of Neural Turing Machines (NTM) in a way similar to LSTMCell.

This repository does not include any training examples. It only includes a simple comparison between torch.nn.LSTMCell and NTMCell implemented here.

Much of the code used here is inspired by this repository.