pytorch/torch/nn
2024-03-08 19:17:20 +00:00
..
attention Make CausalBias a torch.Tensor subclass again (#121358) 2024-03-07 05:20:47 +00:00
backends
intrinsic
modules Bugfix consume_prefix_in_state_dict_if_present function to keep the order of the state_dict (#117464) 2024-03-07 04:00:49 +00:00
parallel [DDP][PT2D] Ignore gradient sync if the gradient is not defined (#120419) 2024-02-29 00:27:54 +00:00
qat
quantizable
quantized
utils Add complex support to parametrizations.spectral_norm (#121452) 2024-03-08 19:17:20 +00:00
__init__.py Add Support for CausalBias to torch compile (#116071) 2024-01-30 02:22:48 +00:00
_reduction.py
common_types.py
cpp.py
functional.py Update docs to point to new sdpa_kernel context manager (#121180) 2024-03-05 22:19:48 +00:00
functional.pyi.in Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
grad.py
init.py
parameter.py Add guardrails preventing complex params in LBFGS & SparseAdam (#118161) 2024-01-24 21:22:47 +00:00
parameter.pyi