PyTorch implementation of QENet