add TorchCompile to callbacks/__init__.py (#718)

Summary:
Pull Request resolved: #718

adds callback to init file

Reviewed By: galrotem

Differential Revision: D54502498

fbshipit-source-id: 6974ae78bb0411398e71e14617904450c6f7f241

torchtnt/framework/callbacks/__init__.py

@@ -15,6 +15,7 @@
 from .pytorch_profiler import PyTorchProfiler
 from .system_resources_monitor import SystemResourcesMonitor
 from .tensorboard_parameter_monitor import TensorBoardParameterMonitor
+from .torch_compile import TorchCompile
 from .torchsnapshot_saver import TorchSnapshotSaver
 from .tqdm_progress_bar import TQDMProgressBar
 from .train_progress_monitor import TrainProgressMonitor
@@ -31,6 +32,7 @@
     "PyTorchProfiler",
     "SystemResourcesMonitor",
     "TensorBoardParameterMonitor",
+    "TorchCompile",
     "TorchSnapshotSaver",
     "TQDMProgressBar",
     "TrainProgressMonitor",