PyTorch implementation for EIE https://arxiv.org/abs/1602.01528 - View it on GitHub