Reference API#
Created On: Apr 16, 2025 | Last Updated On: May 28, 2026
Python API
- torch
- torch.nn
- torch.nn.functional
- torch.Tensor
- Tensor Attributes
- Tensor Views
- torch.amp
- torch.autograd
- torch.library
- torch.accelerator
- torch.cpu
- torch.cuda
- torch.cuda.memory
- torch.mps
- torch.xpu
- torch.mtia
- torch.mtia.memory
- torch.mtia.mtia_graph
- Meta device
- torch.backends
- torch.export
- torch.distributed
- torch.distributed.tensor
- torch.distributed.algorithms.join
- torch.distributed.elastic
- torch.distributed.fsdp
- torch.distributed.fsdp.fully_shard
- torch.distributed.tensor.parallel
- torch.distributed.optim
- torch.distributed.pipelining
- torch.distributed._symmetric_memory
- torch.distributed.checkpoint
- torch.distributions
- torch.compiler
- torch.fft
- torch.func
- torch.futures
- torch.fx
- torch.fx.experimental
- torch.hub
- torch.linalg
- torch.monitor
- torch.signal
- torch.special
- torch.overrides
- torch.nativert
- torch.package
- torch.profiler
- torch.nn.init
- torch.nn.attention
- torch.onnx
- torch.optim
- Complex Numbers
- DDP Communication Hooks
- Quantization
- Distributed RPC Framework
- torch.random
- torch.masked
- torch.nested
- torch.Size
- torch.sparse
- torch.Storage
- torch.testing
- torch.utils
- torch.utils.backend_registration
- torch.utils.hooks
- torch.utils.throughput_benchmark
- torch.utils.collect_env
- torch.utils.flop_counter
- torch.utils.hipify.hipify_python
- Benchmark Utils - torch.utils.benchmark
- torch.utils.benchmark.examples
- torch.utils.benchmark.examples.spectral_ops_fuzz_test
- torch.utils.checkpoint
- torch.utils.cpp_extension
- torch.utils.data
- torch.utils.deterministic
- JIT Utils - torch.utils.jit
- torch.utils.jit.log_extract
- torch.utils.dlpack
- torch.utils.mobile_optimizer
- torch.utils.model_zoo
- torch.utils.tensorboard
- torch.utils.module_tracker
- Type Info
- torch.__config__
- torch.__future__
- torch._logging
- Torch Environment Variables