class_docstr: Root operation which trains an AlphaTensor model to learn more
  efficient matrix multiplications.
code: def __init__(self):
      super().__init__()
      self._model = None
      self._optimizer = None
  
      self._build_model_op = BuildModelOp()
      self._build_optimizer_op = BuildOptimizerOp()
      self._load_checkpoint_op = LoadCheckPointOp()
      self._training_op = TrainingOperation()
      self._save_model_op = SaveModelOp()
  
file_path: nebuly/optimization/open_alpha_tensor/open_alpha_tensor/root_op.py
imports: from typing import Tuple, List
  from nebullvm.operations.base import Operation
  from open_alpha_tensor.core.modules.alpha_tensor import AlphaTensorModel
  from open_alpha_tensor.operations.checkpoint_op import LoadCheckPointOp
  from open_alpha_tensor.operations.model_op import BuildModelOp, SaveModelOp, BuildOptimizerOp
  from open_alpha_tensor.operations.training_op import TrainingOperation
label_desc: Use image node_id 1 to create a new TrainAlphaTensorRootOp object from inherited base classes: Operation with example: obj = TrainAlphaTensorRootOp()