PyTorch is secretly a better choice for rapid prototyping than TensorFlow, don't @ me.