Make LazyAwaitable wait_async dynamo traceable

[IvanKobzarev]

Summary: Clients requested to be able to call directly LazyAwaitable._wait_async(a) under compiled region.

LazyAwaitable.wait_async uses torch.fx.node.map_aggregate which is skipped by dynamo => copied inside torchrec, changed immutable_list -> list, immutable_dict -> dict (immutable versions were failing in dynamo).

Using only for compiled path, not to break any BC on eager path.

Differential Revision: D59113305

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D59113305