[aitune_enable_jit_tuning]
torch = aitune.torch.jit.enable:wrapt_auto_enable
