Index

optimus_dl.modules.model_transforms

Modules and Sub-packages

• base: Abstract base class for all model transformations.
• checkpoint: Activation checkpointing (gradient checkpointing) transform using public PyTorch API.
• compile: Configuration for torch.compile model transform.
• config: Base configuration for model transforms.
• distributed: Distributed model transforms for training.
• load_weights: Configuration for load weights model transform.
• tensor_parallel: Tensor Parallelism Transform.