pytorch/torch
2025-05-31 05:32:31 +00:00
..
_awaits
_C Resubmit Remove MemPoolContext (#154042) (#154746) 2025-05-31 01:21:54 +00:00
_C_flatbuffer
_custom_op
_decomp Fix clamp type promotion in inductor decomposition (#154471) 2025-05-28 23:24:25 +00:00
_dispatch
_dynamo [dynamo][guards] Prevent LENGTH guard on nn modules (#154763) 2025-05-31 05:32:31 +00:00
_export [Export] Add math module for deserialization (#154643) 2025-05-30 17:29:25 +00:00
_functorch [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
_higher_order_ops [refactor] move materialize_as_graph to _higher_order_ops/utils.py (#154070) 2025-05-31 00:06:44 +00:00
_inductor [inductor] Add kernel_hash_key to ChoiceCaller (#154470) 2025-05-31 03:09:37 +00:00
_lazy
_library Add torch._C.Tag.needs_contiguous_strides (#152859) 2025-05-08 04:49:59 +00:00
_logging Revert "convert inductor codecache to use getArtifactLogger (#153766)" 2025-05-30 22:20:07 +00:00
_numpy
_prims [dynamic shapes] unbacked safe unsqueeze (#154087) 2025-05-30 01:41:57 +00:00
_prims_common avoid sym_max on nested int in is_contiguous. (#154633) 2025-05-30 09:59:33 +00:00
_refs introduce definitely_contiguous and use it for reshape and tensor meta data computation. (#153432) 2025-05-28 03:41:26 +00:00
_strobelight
_subclasses Revert "[draft export] avoid storing intermediate real tensors in proxies (#154630)" 2025-05-31 00:07:56 +00:00
_vendor
accelerator
amp
ao remove allow-untyped-defs from torch/ao/quantization/stubs.py (#154622) 2025-05-30 07:26:09 +00:00
autograd Add memory reporting for XPU to Memory Profiler (#152842) 2025-05-21 01:19:19 +00:00
backends Revert "refine fp32 precision api (#125888)" 2025-05-11 00:35:46 +00:00
compiler Support unbacked whitelist (#154295) 2025-05-28 23:01:22 +00:00
contrib
cpu [device_mesh] improve device selection logic (#150897) 2025-05-14 06:29:16 +00:00
csrc Resubmit Remove MemPoolContext (#154042) (#154746) 2025-05-31 01:21:54 +00:00
cuda Resubmit Remove MemPoolContext (#154042) (#154746) 2025-05-31 01:21:54 +00:00
distributed [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
distributions Type hints for distributions/utils (#154712) 2025-05-30 15:50:31 +00:00
export [export] Move PT2ArchiveWriter/Reader to torch/export (#153795) 2025-05-23 19:04:36 +00:00
fft
func
futures Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
fx Revert "[draft export] avoid storing intermediate real tensors in proxies (#154630)" 2025-05-31 00:07:56 +00:00
jit
legacy
lib Revert "Use 3.27 as the minimum CMake version (#153153)" 2025-05-31 02:14:24 +00:00
linalg
masked Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
monitor
mps
mtia Add getDeviceProperties api to torch mtia device (#153577) 2025-05-27 11:55:58 +00:00
multiprocessing
nativert [nativert] move OpKernelKind enum to torch (#154756) 2025-05-31 01:31:29 +00:00
nested Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
nn Fix load_state_dict description (#154599) 2025-05-30 18:08:59 +00:00
onnx [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
optim Fix lr_scheduler unexpectedly calls step() when init argument last_epoch is larger than -1 (#149312) 2025-05-22 08:42:37 +00:00
package [BE]: Enable ruff YTT linter for Python version checks (#153547) 2025-05-14 21:09:16 +00:00
profiler
quantization
signal
sparse Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
special Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
testing [Reland][pytorch] Patch the _is_conv_node function (#154473) 2025-05-30 00:41:03 +00:00
utils ROCm Sparsity through HipSparseLT (#150578) 2025-05-31 02:03:40 +00:00
xpu
__config__.py
__future__.py
__init__.py [Dynamo] Introduce hook receiving list of traced code objects (#153622) 2025-05-28 15:40:09 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_deploy.py
_environment.py
_guards.py [Dynamo] Introduce hook receiving list of traced code objects (#153622) 2025-05-28 15:40:09 +00:00
_jit_internal.py BE: Type previously untyped decorators (#154515) 2025-05-29 00:36:34 +00:00
_linalg_utils.py
_lobpcg.py Fixed rerr computation in lobpcg (#152789) 2025-05-08 12:22:31 +00:00
_lowrank.py
_meta_registrations.py [Intel GPU][Inductor] Fallback embedding_dense_backward on XPU (#151637) 2025-05-19 02:19:37 +00:00
_namedtensor_internals.py
_ops.py Revert "Improve torch.ops typing (#153558)" 2025-05-19 23:32:36 +00:00
_python_dispatcher.py
_size_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_tensor_str.py
_tensor.py
_thread_safe_fork.py
_torch_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_utils_internal.py [reland] Add graph module runtime asserts to AOTI (#153182) 2025-05-09 22:56:19 +00:00
_utils.py Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
CMakeLists.txt Revert "Use 3.27 as the minimum CMake version (#153153)" 2025-05-31 02:14:24 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
header_only_apis.txt Add torch/header_only_apis.txt and enforce they're tested (#153635) 2025-05-20 23:42:24 +00:00
hub.py
library.h
library.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
overrides.py
py.typed
quasirandom.py
random.py
return_types.py
script.h
serialization.py Update serialization docs (#153631) 2025-05-19 20:22:07 +00:00
storage.py
torch_version.py
types.py
version.py.tpl