Amazon SageMaker

TODO: Document SageMaker integration.