Pin torch thread counts in unit tests and revert CPU bump