Bump torch to <2.4.1

[eitanturok]

Update: Running the latest tests manually and they pass here

What does this PR do?

• use "torch >= 2.4.0, < 2.4.1" in setup.py, pyproject.toml
• use forks of grouped_gemm, stk that support torch2.4
  • will update to official versions of groupped_gemm, stk once they are released
• use newer version of composer that supports torch2.4
  • will use official version of composer once it is released
• tests use torch2.4 image

This Github Actions here run the old tests and predictably fail. I manually ran the new Github Actions here and they all pass.

What issue(s) does this change relate to?

Before submitting

• Have you read the contributor guidelines?
• Is this change a documentation change or typo fix? If so, skip the rest of this checklist.
• Was this change discussed/approved in a GitHub issue first? It is much more likely to be merged if so.
• Did you update any related docs and document your change?
• Did you update any related tests and add any new tests related to your change? (see testing)
• Did you run the tests locally to make sure they pass?
• Did you run pre-commit on your change? (see the pre-commit section of prerequisites)

[eitanturok]

blocked by waiting for new releases of composer, groupped_gemm, stk
resolved

[snarayan21]

compsoer version

[snarayan21]

lgtm

[eitanturok]

Manually ran the tests again and they are failing here

[eitanturok]

Running tests manually and the pass here