DISCONTINUED: I've stopped working on this halfway through. THIS DOES NOT WORK.
Implementation of the A2OC algorithm in PyTorch