← all repositories

loudinthecloud/pytorch-ntm

A PyTorch implementation of Neural Turing Machines with external memory for learning algorithmic tasks.

610 stars Jupyter Notebook ML Frameworks
pytorch-ntm
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository provides a vanilla implementation of Neural Turing Machines in PyTorch. An NTM extends traditional sequence processors (like LSTMs) with an external differentiable memory bank, allowing the network to learn algorithmic tasks and maintain internal state longer than standard recurrent networks. The implementation supports batch learning, flexible read/write head configurations, and includes copy and repeat-copy task experiments that reproduce results from the original NTM paper.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.