transformers

976d8cc9 - Follow-up typing checking fixes (#44500)

Follow-up typing checking fixes (#44500) - Do proper type check in case jax is installed. - Make sure older torch versions don't raise typing issues