A PyTorch plugin to rethink AI's core