Upgrade PyTorch to 2.0.1

[namannandan]

Description

Upgrade the following packages:
torch == 2.0.1
torchvision == 0.15.2
torchtext == 0.15.2
torchaudio == 2.0.2

Type of change

• Breaking change (fix or feature that would cause existing functionality to not work as expected)

Feature/Issue validation/testing

• Sanity and regression tests in CI

[codecov]

Codecov Report

Merging #2374 (a0f7193) into master (e5004b2) will not change coverage.
The diff coverage is n/a.

❗ Current head a0f7193 differs from pull request most recent head 37a5e10. Consider uploading reports for the commit 37a5e10 to get more accurate results

@@           Coverage Diff           @@
##           master    #2374   +/-   ##
=======================================
  Coverage   72.13%   72.13%           
=======================================
  Files          78       78           
  Lines        3642     3642           
  Branches       58       58           
=======================================
  Hits         2627     2627           
  Misses       1011     1011           
  Partials        4        4           

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

[namannandan]

Pending validation of successful regression test runs after ongoing fixes to regression test suite