Amazon SageMaker now supports Pipe Input Mode for the built-in TensorFlow containers. Pipe Input Mode enables data to stream directly from Amazon Simple Storage Service (Amazon S3) to the TensorFlow container on the training instance, using the TensorFlow dataset construct.