Write your own training loop — without managing GPUs — with Amazon SageMaker training session APIs | AIChainDay