All modules for which code is available
- torch
- torch._C
- torch._C._distributed_autograd
- torch._C._distributed_c10d
- torch._C._distributed_rpc
- torch._C._fft
- torch._C._linalg
- torch._C._monitor
- torch._C._nested
- torch._C._nn
- torch._C._profiler
- torch._C._sparse
- torch._C._special
- torch.__config__
- torch.__future__
- torch._jit_internal
- torch._lobpcg
- torch._logging._internal
- torch._lowrank
- torch._tensor
- torch._tensor_str
- torch._utils
- torch.amp.autocast_mode
- torch.ao.nn.intrinsic.modules.fused
- torch.ao.nn.intrinsic.qat.modules.conv_fused
- torch.ao.nn.intrinsic.qat.modules.linear_relu
- torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu
- torch.ao.nn.intrinsic.quantized.modules.bn_relu
- torch.ao.nn.intrinsic.quantized.modules.conv_relu
- torch.ao.nn.intrinsic.quantized.modules.linear_relu
- torch.ao.nn.qat.dynamic.modules.linear
- torch.ao.nn.qat.modules.conv
- torch.ao.nn.qat.modules.linear
- torch.ao.nn.quantizable.modules.activation
- torch.ao.nn.quantizable.modules.rnn
- torch.ao.nn.quantized.dynamic.modules.linear
- torch.ao.nn.quantized.dynamic.modules.rnn
- torch.ao.nn.quantized.functional
- torch.ao.nn.quantized.modules.activation
- torch.ao.nn.quantized.modules.batchnorm
- torch.ao.nn.quantized.modules.conv
- torch.ao.nn.quantized.modules.embedding_ops
- torch.ao.nn.quantized.modules.functional_modules
- torch.ao.nn.quantized.modules.linear
- torch.ao.nn.quantized.modules.normalization
- torch.ao.ns._numeric_suite
- torch.ao.ns._numeric_suite_fx
- torch.ao.ns.fx.utils
- torch.ao.quantization
- torch.ao.quantization.backend_config.backend_config
- torch.ao.quantization.fake_quantize
- torch.ao.quantization.fuse_modules
- torch.ao.quantization.fx.custom_config
- torch.ao.quantization.observer
- torch.ao.quantization.pt2e.export_utils
- torch.ao.quantization.qconfig
- torch.ao.quantization.qconfig_mapping
- torch.ao.quantization.quantize
- torch.ao.quantization.quantize_fx
- torch.ao.quantization.stubs
- torch.autograd
- torch.autograd.anomaly_mode
- torch.autograd.forward_ad
- torch.autograd.function
- torch.autograd.functional
- torch.autograd.grad_mode
- torch.autograd.gradcheck
- torch.autograd.graph
- torch.autograd.profiler
- torch.autograd.profiler_util
- torch.backends.cpu
- torch.backends.cuda
- torch.backends.cudnn
- torch.backends.mha
- torch.backends.mkl
- torch.backends.mkldnn
- torch.backends.mps
- torch.backends.nnpack
- torch.backends.openmp
- torch.backends.opt_einsum
- torch.compiler
- torch.cpu
- torch.cuda
- torch.cuda._sanitizer
- torch.cuda.amp.autocast_mode
- torch.cuda.amp.grad_scaler
- torch.cuda.graphs
- torch.cuda.jiterator
- torch.cuda.memory
- torch.cuda.nvtx
- torch.cuda.random
- torch.cuda.streams
- torch.cuda.tunable
- torch.distributed
- torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks
- torch.distributed.algorithms.ddp_comm_hooks.default_hooks
- torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook
- torch.distributed.algorithms.join
- torch.distributed.autograd
- torch.distributed.checkpoint.default_planner
- torch.distributed.checkpoint.filesystem
- torch.distributed.checkpoint.format_utils
- torch.distributed.checkpoint.planner
- torch.distributed.checkpoint.staging
- torch.distributed.checkpoint.state_dict
- torch.distributed.checkpoint.state_dict_loader
- torch.distributed.checkpoint.state_dict_saver
- torch.distributed.checkpoint.stateful
- torch.distributed.checkpoint.storage
- torch.distributed.device_mesh
- torch.distributed.distributed_c10d
- torch.distributed.elastic.agent.server.api
- torch.distributed.elastic.agent.server.health_check_server
- torch.distributed.elastic.agent.server.local_elastic_agent
- torch.distributed.elastic.control_plane
- torch.distributed.elastic.events
- torch.distributed.elastic.metrics.api
- torch.distributed.elastic.multiprocessing
- torch.distributed.elastic.multiprocessing.api
- torch.distributed.elastic.multiprocessing.errors
- torch.distributed.elastic.multiprocessing.subprocess_handler.handlers
- torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler
- torch.distributed.elastic.rendezvous.api
- torch.distributed.elastic.rendezvous.c10d_rendezvous_backend
- torch.distributed.elastic.rendezvous.dynamic_rendezvous
- torch.distributed.elastic.rendezvous.etcd_rendezvous
- torch.distributed.elastic.rendezvous.etcd_rendezvous_backend
- torch.distributed.elastic.rendezvous.etcd_server
- torch.distributed.elastic.rendezvous.etcd_store
- torch.distributed.elastic.timer.api
- torch.distributed.elastic.timer.debug_info_logging
- torch.distributed.elastic.timer.file_based_local_timer
- torch.distributed.elastic.timer.local_timer
- torch.distributed.fsdp.api
- torch.distributed.fsdp.fully_sharded_data_parallel
- torch.distributed.nn.api.remote_module
- torch.distributed.optim.optimizer
- torch.distributed.optim.post_localSGD_optimizer
- torch.distributed.optim.zero_redundancy_optimizer
- torch.distributed.pipelining._IR
- torch.distributed.pipelining.microbatch
- torch.distributed.pipelining.schedules
- torch.distributed.pipelining.stage
- torch.distributed.rpc
- torch.distributed.rpc.api
- torch.distributed.rpc.backend_registry
- torch.distributed.rpc.functions
- torch.distributed.rpc.options
- torch.distributed.tensor.parallel.api
- torch.distributed.tensor.parallel.loss
- torch.distributed.tensor.parallel.style
- torch.distributions.bernoulli
- torch.distributions.beta
- torch.distributions.binomial
- torch.distributions.categorical
- torch.distributions.cauchy
- torch.distributions.chi2
- torch.distributions.constraint_registry
- torch.distributions.constraints
- torch.distributions.continuous_bernoulli
- torch.distributions.dirichlet
- torch.distributions.distribution
- torch.distributions.exp_family
- torch.distributions.exponential
- torch.distributions.fishersnedecor
- torch.distributions.gamma
- torch.distributions.geometric
- torch.distributions.gumbel
- torch.distributions.half_cauchy
- torch.distributions.half_normal
- torch.distributions.independent
- torch.distributions.inverse_gamma
- torch.distributions.kl
- torch.distributions.kumaraswamy
- torch.distributions.laplace
- torch.distributions.lkj_cholesky
- torch.distributions.log_normal
- torch.distributions.lowrank_multivariate_normal
- torch.distributions.mixture_same_family
- torch.distributions.multinomial
- torch.distributions.multivariate_normal
- torch.distributions.negative_binomial
- torch.distributions.normal
- torch.distributions.one_hot_categorical
- torch.distributions.pareto
- torch.distributions.poisson
- torch.distributions.relaxed_bernoulli
- torch.distributions.relaxed_categorical
- torch.distributions.studentT
- torch.distributions.transformed_distribution
- torch.distributions.transforms
- torch.distributions.uniform
- torch.distributions.utils
- torch.distributions.von_mises
- torch.distributions.weibull
- torch.distributions.wishart
- torch.export
- torch.export.dynamic_shapes
- torch.export.exported_program
- torch.export.graph_signature
- torch.export.unflatten
- torch.func
- torch.functional
- torch.futures
- torch.fx._symbolic_trace
- torch.fx.experimental.symbolic_shapes
- torch.fx.graph
- torch.fx.graph_module
- torch.fx.interpreter
- torch.fx.node
- torch.fx.proxy
- torch.fx.subgraph_rewriter
- torch.hub
- torch.jit
- torch.jit._async
- torch.jit._freeze
- torch.jit._fuser
- torch.jit._script
- torch.jit._serialization
- torch.jit._trace
- torch.library
- torch.monitor
- torch.mps
- torch.mtia
- torch.multiprocessing
- torch.nested
- torch.nn.attention
- torch.nn.functional
- torch.nn.init
- torch.nn.modules.activation
- torch.nn.modules.adaptive
- torch.nn.modules.batchnorm
- torch.nn.modules.channelshuffle
- torch.nn.modules.container
- torch.nn.modules.conv
- torch.nn.modules.distance
- torch.nn.modules.dropout
- torch.nn.modules.flatten
- torch.nn.modules.fold
- torch.nn.modules.instancenorm
- torch.nn.modules.lazy
- torch.nn.modules.linear
- torch.nn.modules.loss
- torch.nn.modules.module
- torch.nn.modules.normalization
- torch.nn.modules.padding
- torch.nn.modules.pixelshuffle
- torch.nn.modules.pooling
- torch.nn.modules.rnn
- torch.nn.modules.sparse
- torch.nn.modules.transformer
- torch.nn.modules.upsampling
- torch.nn.parallel.comm
- torch.nn.parallel.data_parallel
- torch.nn.parallel.distributed
- torch.nn.parameter
- torch.nn.utils.clip_grad
- torch.nn.utils.convert_parameters
- torch.nn.utils.fusion
- torch.nn.utils.init
- torch.nn.utils.memory_format
- torch.nn.utils.parametrizations
- torch.nn.utils.parametrize
- torch.nn.utils.prune
- torch.nn.utils.rnn
- torch.nn.utils.spectral_norm
- torch.nn.utils.stateless
- torch.nn.utils.weight_norm
- torch.onnx
- torch.optim.adadelta
- torch.optim.adagrad
- torch.optim.adam
- torch.optim.adamax
- torch.optim.adamw
- torch.optim.asgd
- torch.optim.lbfgs
- torch.optim.lr_scheduler
- torch.optim.nadam
- torch.optim.optimizer
- torch.optim.radam
- torch.optim.rmsprop
- torch.optim.rprop
- torch.optim.sgd
- torch.optim.sparse_adam
- torch.overrides
- torch.package.file_structure_representation
- torch.package.package_exporter
- torch.package.package_importer
- torch.profiler.itt
- torch.profiler.profiler
- torch.quasirandom
- torch.random
- torch.serialization
- torch.signal.windows.windows
- torch.sparse
- torch.storage
- torch.testing._comparison
- torch.testing._creation
- torch.utils
- torch.utils.backend_registration
- torch.utils.benchmark.utils.common
- torch.utils.benchmark.utils.compare
- torch.utils.benchmark.utils.timer
- torch.utils.benchmark.utils.valgrind_wrapper.timer_interface
- torch.utils.checkpoint
- torch.utils.cpp_backtrace
- torch.utils.cpp_extension
- torch.utils.data._utils.collate
- torch.utils.data._utils.worker
- torch.utils.data.dataloader
- torch.utils.data.dataset
- torch.utils.data.distributed
- torch.utils.data.sampler
- torch.utils.dlpack
- torch.utils.mobile_optimizer
- torch.utils.module_tracker
- torch.utils.tensorboard.writer
- torch.xpu
- typing