Some types for remote_module (#58012)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58012
Test Plan: Sandcastle
Reviewed By: SciPioneer
Differential Revision: D28334611
fbshipit-source-id: 5e4645a7de65e064cb6a919cdc2372151ec48d44
