PyTorch is still the only reasonable choice for rapid prototyping, don't @ me