Support Training/Deployment on AWS SageMaker

[pm3310]

I'd like to start using tensorfx but I'd like to train tensorfx models on SageMaker. Additionally, I'd like to deploy trained tensorfx models on SageMaker as REST Endpoints.

I have open sourced https://github.com/Kenza-AI/sagify, a CLI tool (it can be used as a library, too) that enables the training and deployment of any custom ML pipeline. Example of CLI command to train on SageMaker:

sagify cloud train -d src/ -i s3://my-bucket/training-data/ -o s3://my-bucket/output/ -e ml.m4.xlarge

So, I propose something like that:

tfx sagemaker train -d src/ -i s3://my-bucket/training-data/ -o s3://my-bucket/output/ -e ml.m4.xlarge