1. Handle None as inputs 2. Turn the function into `trace_only` 3. Fix failing CI tests Ref: - https://github.com/pytorch/pytorch/issues/131349 - https://github.com/pytorch/pytorch/issues/119363