pytorch/torch/nn
2024-11-06 18:20:57 +00:00
..
attention [Flex Attention] Paged Attention (#137164) 2024-10-29 17:05:22 +00:00
backends
intrinsic
modules [pytorch] Make global module hook to pass kwargs similar to how module hook works (#137403) 2024-11-06 18:20:57 +00:00
parallel Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-11-05 23:10:38 +00:00
qat
quantizable
quantized
utils
__init__.py
_reduction.py
common_types.py
cpp.py
functional.py Add Weighted Loss Functions to PyTorch : WMSE, WMAE, and Weighted Huber Loss (#132049) 2024-10-31 21:59:43 +00:00
functional.pyi.in
grad.py
init.py
parameter.py
parameter.pyi