All modules for which code is available

• builtins
• collections
• tensorplay
• tensorplay._C
• tensorplay._composite_funcs
• tensorplay._einsum
• tensorplay._finfo
• tensorplay._shape_funcs
• tensorplay._stax.api
• tensorplay._stax.stax
• tensorplay.amp.autocast_mode
• tensorplay.amp.grad_scaler
• tensorplay.autograd
• tensorplay.autograd.anomaly_mode
• tensorplay.autograd.forward_ad
• tensorplay.autograd.function
• tensorplay.autograd.functional
• tensorplay.autograd.grad_mode
• tensorplay.autograd.gradcheck
• tensorplay.autograd.graph
• tensorplay.autograd.profiler_util
• tensorplay.autograd.variable
• tensorplay.cuda
• tensorplay.cuda._utils
• tensorplay.cuda.gds
• tensorplay.cuda.graph_annotations
• tensorplay.cuda.graphs
• tensorplay.cuda.memory
• tensorplay.cuda.nccl
• tensorplay.cuda.nvtx
• tensorplay.cuda.profiler
• tensorplay.cuda.random
• tensorplay.cuda.streams
• tensorplay.distributed.distributed_core
• tensorplay.fft._frequency
• tensorplay.fft._transforms
• tensorplay.functional
• tensorplay.futures
• tensorplay.library
• tensorplay.nn.functional
• tensorplay.nn.init
• tensorplay.nn.modules.flatten
• tensorplay.nn.modules.lazy
• tensorplay.nn.modules.module
• tensorplay.nn.parameter
• tensorplay.nn.utils.rnn
• tensorplay.nn.utils.stateless
• tensorplay.profiler.profiler
• tensorplay.random
• tensorplay.serialization.api
• tensorplay.serialization.archive
• tensorplay.serialization.policy
• tensorplay.special._elementwise
• tensorplay.special._error
• tensorplay.special._gamma
• tensorplay.stax
• tensorplay.utils.checkpoint
• tensorplay.utils.data