Minor update to the pytorch version documentation, and a little extra info for the pytorch benchmarks.