pytorch 2.0 test failures

[ndmlny-qs]

Issue Description

Need to pin pytorch to 1.13.1 so tests pass.

Steps to Reproduce

Expected Behavior

System Info

Please provide information about your setup

• PyTorch Version (run print(torch.__version__)
• Python version

Additional Context

[zaxtax]

@horizon-blue what approach did you think is best to quickly make things work?