do small edge boxes make sense with PyTorch

[emberpeak]

My first guess would be this: Start by verifying the driver/runtime combo, then test with a smaller prompt, then look at continuous batching.