Implementation of the Value Iteration Network in Pytorch - View it on GitHub