Remove typ for torch upper version

[harimkang]

Summary

#3766
I simply checked classification training testing with torch==2.3.1 and it should work with that modification.

How to test

Checklist

• I have added unit tests to cover my changes.​
• I have added integration tests to cover my changes.​
• I have ran e2e tests and there is no issues.
• I have added the description of my changes into CHANGELOG in my target branch (e.g., CHANGELOG in develop).​
• I have updated the documentation in my target branch accordingly (e.g., documentation in develop).
• I have linked related issues.

License

• I submit my code changes under the same Apache License that covers the project.
  Feel free to contact the maintainers if that's a concern.
• I have updated the license header for each file (see an example below).

# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

[eunwoosh]

We need to consider XPU support also. Currently, latest torch version IPEX supports is 2.1.3
I requested change to prevent it from merging before dealing with it.

[harimkang]

We need to consider XPU support also. Currently, latest torch version IPEX supports is 2.1.3 I requested change to prevent it from merging before dealing with it.

This is not a PR to upgrade the torch version, it's just a PR to prepare for future changes to _register_pytree_node. Fix it so that it works the same in the current and future versions.

[eunwoosh]

We need to consider XPU support also. Currently, latest torch version IPEX supports is 2.1.3 I requested change to prevent it from merging before dealing with it.

This is not a PR to upgrade the torch version, it's just a PR to prepare for future changes to _register_pytree_node. Fix it so that it works the same in the current and future versions.

Sorry, I misunderstood it as upgrading torch version. Thanks for explanation :)