Hi, sorry for the delay responding!  Sayak you are correct that TFX offers some deeper support for NLP in the Evaluator component, by offering BLEU and ROUGE metrics.
For image processing we don’t yet have the same level of support, but there is an example 8 for this.  It shows:

How to use a Keras application and to fine-tune the MobileNet image classification model.
How to perform data augmentation within the pipeline to ensure that the model generalizes well.
It invokes the rewriter to convert the model into TFLite.
Finally it shows how to write the required TFLite metadata to make the model compatible with MLKit.