For AI agents: a documentation index is available at the root level at /llms.txt. Append /llms.txt to any URL for a page-level index, or .md for the markdown version of any page.
LogoLogoDocumentation
  • Learn
    • Overview
    • Install
    • Quick Start
    • Core Functionalities
    • Backends
    • Observability
  • Guides
    • Ahead-of-time Tuning
    • Ahead-of-time Inspect
    • Just-in-time Tuning
    • Just-in-time Inspect
    • Tune Strategies
    • Deployment
  • Examples
    • Overview
    • E5Large
    • ESM2
    • FLUX
    • JitTuning
    • LLM
    • ParakeetCTC
    • ParakeetRNNT
    • ResNet
    • StableDiffusion
  • API Reference
      • Aitune
        • Dynamo
        • Exceptions
        • Global Context
        • Torch
          • Backend
          • Checkpoint
            • Checkpoint
            • Local Torch Storage
            • Storage
            • Storage Tasks
            • Torch Checkpoint
          • Config
          • Dataloader
          • Dynamic Shapes
          • Inspecting
          • Integrations
          • Jit
          • Libs
          • Module
          • Module Registry
          • Performance
          • Task
          • Tune Strategy
          • Tuning
        • Utils
  • Resources
    • Changelog
    • Known Issues
    • Contributing
    • License
  • Overview
  • Install
  • Quick Start
  • Core Functionalities
  • Backends
  • Observability
  • Ahead-of-time Tuning
  • Ahead-of-time Inspect
  • Just-in-time Tuning
  • Just-in-time Inspect
  • TensorRT Backend
  • TensorRT Optimization Profiles
  • ONNXRuntime Backend
  • Torch-TensorRT Backend (JIT)
  • Torch-TensorRT Backend (AOT)
  • TorchAO Backend
  • Torch Inductor JIT Backend
  • Torch Inductor AOT Backend
  • Tune Strategies
  • Deployment
  • Tuning Workflow
  • Execution Graphs
  • Profiling and Hardware Metrics
  • Performance Profile
  • Tuning Assistant
  • Overview
  • E5Large
  • ESM2
  • FLUX
  • JitTuning
  • LLM
  • ParakeetCTC
  • ParakeetRNNT
  • ResNet
  • StableDiffusion
  • Module
  • Tuning
  • Config
  • Backend
  • Inspect
  • Config
  • TensorRT Backend
  • Torch-TensorRT Backend (JIT)
  • Torch-TensorRT Backend (AOT)
  • TorchAO Backend
  • Torch Inductor JIT Backend
  • Torch Inductor AOT Backend
  • FirstWinsStrategy
  • OneBackendStrategy
  • MaxThroughputStrategy
  • LatencyBudgetStrategy
  • MinLatencyStrategy
  • Aitune
  • Dynamo
  • Worker
  • Exceptions
  • Global Context
  • Torch
  • Backend
  • Backend
  • Onnx Runtime Backend
  • Tensorrt
  • Modelopt Calibration
  • Onnx Autocast
  • Onnx Quantization
  • Tensorrt Backend
  • Tensorrt Builder
  • Tensorrt Engine Info
  • Tensorrt Profile
  • Tensorrt Runtime
  • Torch Output Allocator
  • Torch Quantization
  • Torch Eager
  • Torch Inductor Aot Backend
  • Torch Inductor Jit Backend
  • Torch Tensorrt Aot Backend
  • Torch Tensorrt Jit Backend
  • Torchao Backend
  • Checkpoint
  • Checkpoint
  • Local Torch Storage
  • Storage
  • Storage Tasks
  • Torch Checkpoint
  • Config
  • Dataloader
  • Dynamic Shapes
  • Inspecting
  • Inspecting
  • Module Info
  • Module Inspector
  • Wrapping
  • Integrations
  • Diffusers
  • Transformers
  • Jit
  • Config
  • Disable
  • Enable
  • Enable Inspection
  • Html Generator
  • Inspect Module
  • Patched Module
  • Patcher
  • Tune
  • Libs
  • Cuda
  • Cudart
  • Memory
  • Onnx
  • Onnx Exporter
  • Onnx Model Info
  • Torch Compile
  • Module
  • Forward Signature
  • Graph Spec
  • Locator
  • Passthrough Module
  • Recording Module
  • Sample Metadata
  • Tensor Spec
  • Tuned Module
  • Wrapper Module
  • Module Registry
  • Performance
  • Attribution Hooks
  • Context
  • Markdown
  • Profile
  • Runtime Profile
  • Utils
  • Task
  • Correctness
  • Find Max Batch Size
  • Profiling
  • Config
  • Events
  • Measuring Stop Strategy
  • Measuring Strategy
  • Metrics
  • Profiling
  • Profiling Stop Strategy
  • Tune Strategy
  • First Wins Strategy
  • Latency Budget Strategy
  • Max Throughput Strategy
  • Min Latency Strategy
  • Mixin
  • Find Max Batch Size Mixin
  • Performance Validation Mixin
  • One Backend Strategy
  • Profiling Tune Strategy
  • Tune Strategy
  • Tuning
  • Utils
  • Disk Space
  • Env Vars
  • Hashing
  • Logging
  • Monitoring
  • Annotation
  • Context
  • Backend Context
  • Hardware Metrics
  • Child Receiver
  • Collector
  • Ipc
  • Metrics
  • Parent Receiver
  • Hardware Metrics Annotation
  • Nvtx Annotation
  • Parse Metrics
  • Setup Hardware Metrics
  • Serialization
  • Timer
  • Validation
  • Changelog
  • Known Issues
  • Contributing
  • License
On this page
  • Submodules
API ReferenceOther APIsAituneTorch

aitune.torch.checkpoint

||View as Markdown|

Checkpoint module.

Submodules

  • aitune.torch.checkpoint.checkpoint
  • aitune.torch.checkpoint.local_torch_storage
  • aitune.torch.checkpoint.storage
  • aitune.torch.checkpoint.storage_tasks
  • aitune.torch.checkpoint.torch_checkpoint
Previous

aitune.torch.backend.torch_eager

Next

aitune.torch.checkpoint.checkpoint

NVIDIANVIDIA
Developer-friendly docs for your API
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2026, NVIDIA Corporation.