Torch version

Can we use this project with torch2.1.0 or other version? <!-- POLAR PLEDGE BADGE START --> ## Upvote & Fund - We're using [Polar.sh](https://polar.sh/kyegomez) so you can upvote and help fund this issue. - We receive the funding once the issue is completed & confirmed by you. - Thank you in advance for helping prioritize & fund our backlog. <a href="https://polar.sh/kyegomez/Jamba/issues/9"> <picture> <source media="(prefers-color-scheme: dark)" srcset="https://polar.sh/api/github/kyegomez/Jamba/issues/9/pledge.svg?darkmode=1"> <img alt="Fund with Polar" src="https://polar.sh/api/github/kyegomez/Jamba/issues/9/pledge.svg"> </picture> </a> <!-- POLAR PLEDGE BADGE END -->