aitune.torch.jit.enable | NVIDIA AITune Documentation

Module for enabling JIT tuning.

If you import this module, all torch models will be tuned automatically.

Module Contents

Name	Description
`wrapt_auto_enable`	Entry point for enabling JIT tuning by autowrapt.

aitune.torch.jit.enable.wrapt_auto_enable(
    _
)

Entry point for enabling JIT tuning by autowrapt.