move misc implementation out of `jit/__init__.py` (#41154)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41154
Test Plan: Imported from OSS
Reviewed By: ailzhang
Differential Revision: D22445213
Pulled By: suo
fbshipit-source-id: 200545715c5ef13beb1437f49e01efb21498ddb7
