"stage","description"
"Training","tf.functions can be used to improve the training speed of custom Keras models. Tf.function is commonly used to speed up training loops."
"Data preparation","In this step, we load data in a suitable place and prepare it for use for ML training. "
"Prediction","Use tf.function to speed up the prediction."
"None","Not in the ML/DL pipeline"
"Unknown","Could not identify the ML/DL pipeline"