PyTorch is still the superior choice for rapid prototyping, don't @ me.