pytorch/torch/nn
2025-07-16 06:06:29 +00:00
..
attention Add better typing to avaialbe kernel options for flex attention (#158383) 2025-07-16 06:06:29 +00:00
backends
intrinsic
modules FractionalMaxPool3d add kernel_size check (#155549) 2025-07-10 04:55:06 +00:00
parallel XCCL changes for DDP (#155497) 2025-07-03 05:18:08 +00:00
qat [BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548) 2025-06-14 11:27:04 +00:00
quantizable [BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548) 2025-06-14 11:27:04 +00:00
quantized [BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548) 2025-06-14 11:27:04 +00:00
utils [BE][Ez]: Fully type nn.utils.clip_grad (#154801) 2025-07-09 14:27:51 +00:00
__init__.py
_reduction.py
common_types.py [BE][Ez]: Denote common types as TypeAlias (#154527) 2025-05-29 02:00:13 +00:00
cpp.py
functional.py [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
functional.pyi.in fix type hints for interpolation functions (#157202) 2025-07-09 03:11:37 +00:00
grad.py
init.py [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
parameter.py
parameter.pyi