List of all items

Structs

• async_collective::PendingCollective
• backend::SimulatedBackend
• backend::SubBackend
• backend::TcpBackend
• checkpoint::AsyncCheckpointer
• checkpoint::CheckpointFuture
• checkpoint::DistributedCheckpoint
• checkpoint::ShardMetadata
• checkpoint::TensorShardSpec
• ddp::DDP
• device_mesh::DeviceMesh
• dtensor::DTensor
• fsdp::FSDP
• gloo_backend::GlooBackend
• mpi_backend::MpiBackend
• pipeline::Pipeline
• rpc::RpcAgent
• rpc::TcpRpcBackend
• sync_batch_norm::SyncBatchNorm2d
• ucc_backend::UccBackend

Enums

• checkpoint::DistCheckpointError
• collective::ReduceOp
• dtensor::Placement
• error::DistributedError
• fsdp::ShardingStrategy
• pipeline::PipelineSchedule
• rpc::RpcError

Traits

• backend::Backend

Functions

• async_collective::async_all_gather
• async_collective::async_reduce_scatter
• checkpoint::flat_shard_metadata
• checkpoint::load_distributed
• checkpoint::reshard
• checkpoint::save_distributed
• collective::all_gather
• collective::all_gather_with_timeout
• collective::all_to_all
• collective::all_to_all_single_uneven
• collective::all_to_all_with_timeout
• collective::allreduce
• collective::allreduce_with_timeout
• collective::barrier
• collective::broadcast
• collective::reduce_scatter
• collective::reduce_scatter_tensor
• collective::reduce_scatter_with_timeout
• gloo_backend::is_gloo_available
• mpi_backend::is_mpi_available
• p2p::recv
• p2p::recv_into
• p2p::recv_into_with_timeout
• p2p::recv_with_timeout
• p2p::send
• p2p::sendrecv
• ucc_backend::is_ucc_available

Constants

• collective::DEFAULT_COLLECTIVE_TIMEOUT