Name	Size	Mode	Actions
_autograd.pyi	2744	0644	editdlrm
_cudnn.pyi	378	0644	editdlrm
_distributed_autograd.pyi	908	0644	editdlrm
_distributed_c10d.pyi	9376	0644	editdlrm
_distributed_rpc.pyi	5630	0644	editdlrm
_distributed_rpc_testing.pyi	1136	0644	editdlrm
_functions.pyi	345	0644	editdlrm
_nn.pyi	2050	0644	editdlrm
_nvtx.pyi	150	0644	editdlrm
_onnx.pyi	646	0644	editdlrm
_VariableFunctions.pyi	141026	0644	editdlrm
__init__.pyi	103906	0644	editdlrm

import torch from typing import Dict, List, Set, Any # This module is defined in torch/csrc/distributed/autograd/init.cpp class DistAutogradContext: def _context_id(self) -> int: ... def _recv_functions(self) -> Dict[int, Any]: ... def _send_functions(self) -> Dict[int, Any]: ... def _known_worker_ids(self) -> Set[int]: ... def _new_context() -> DistAutogradContext: ... def _release_context(context_id: int) -> None: ... def _get_max_id() -> int: ... def _is_valid_context(worker_id: int) -> bool: ... def _retrieve_context(context_id: int) -> DistAutogradContext: ... def _current_context() -> DistAutogradContext: ... def _init(worker_id: int) -> None: ... def _get_debug_info() -> Dict[str, str]: ... def backward( context_id: int, roots: List[torch.Tensor], retain_graph = False ) -> None: ... def get_gradients(context_id: int) -> Dict[torch.Tensor, torch.Tensor]: ...