Mahnoo06

Hello, I am using functorch for my project. It's a good substitute for Jax and makes life easy. However it does not have GPU support for M1 chips which is...