Neural Nearest Neighbors Networks (NIPS*2018) - View it on GitHub