Skip to content

unixpickle/lazyseq

Repository files navigation

lazyrnn

This is an API for evaluating and training RNNs on memory-constrained systems. It provides APIs for some of the memory-saving algorithms described in Gruslys et al. and Chen et al..

Why?

Traditional back-propagation requires O(T) memory for sequences of length T. This makes RNNs difficult to train on certain tasks, even when the RNN could theoretically learn the task. One example of such a task is meta-learning with long episodes.

With less naive back-propagation techniques, memory consumption can be reduced to O(log(T)) without significant performance sacrifices. This makes RNNs suitable for a much wider range of applications.

About

Memory-efficient recurrent neural networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages