API documentation

API documentation

• Wrapper
  • CallWrapper
    • CallWrapper.atomic()
    • CallWrapper.disable_hang_protection()
    • CallWrapper.iteration
    • CallWrapper.ping()
  • Wrapper
• Compose
• State
• Rank Assignment
  • Rank Assignment
    • Base class
    • Tree
    • Composable Rank Assignments
  • Rank Filtering
    • Base class
    • Rank Filters
• Rank Filter
• Initialize
  • Initialize
    • Initialize.__call__()
  • RetryController
• Abort
  • Abort
    • Abort.__call__()
  • AbortPersistentCheckpointProcesses
  • AbortTorchDistributed
  • AbortTransformerEngine
• Finalize
  • Finalize
    • Finalize.__call__()
  • ThreadedFinalize
• Health Check
  • HealthCheck
    • HealthCheck.__call__()
  • ChainedGPUHealthCheck
  • ChainedNVLHealthCheck
  • ChainedNicHealthCheck
  • CudaHealthCheck
  • FaultCounter
  • FaultCounterExceeded
  • Enhanced Health Check Features
• Exceptions
  • HealthCheckError
  • InternalError
  • RestartAbort
  • RestartError
  • TimeoutError