Amazon SageMaker training jobs now support ml.p5 instances, powered by NVIDIA H100 chips, which are purpose built for high-performance ML training applications in the cloud. You can use ml.p5 instances on SageMaker to train some of the most demanding models. This includes large language models (LLMs) and diffusion models powering the most demanding generative AI applications. These applications include question answering, code generation, video and image generation, and speech recognition.
Source:: Amazon AWS