Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors - View it on GitHub