Amazon SageMaker Studio notebooks now support P5.4xl instance types
H100 GPUs are now available in SageMaker Studio notebooks, cutting training costs 40% and speeding up LLM and diffusion model work 4x.
View original announcement →Visual Summary
What's New
Amazon EC2 P5.4xl instances, powered by NVIDIA H100 Tensor Core GPUs, are now generally available for use directly within Amazon SageMaker Studio notebooks. This brings one of AWS's most powerful GPU instance types into the interactive notebook environment, enabling data scientists and ML engineers to leverage H100 performance for deep learning and HPC workloads without leaving their development workflow. The availability spans multiple global regions, making high-performance generative AI experimentation and model training more accessible across a broader customer base.
How It Works
- P5.4xl instances are selectable as the compute kernel when launching or switching instance types within SageMaker Studio's JupyterLab or CodeEditor applications, requiring no infrastructure provisioning outside the Studio interface.
- Each P5.4xl instance is backed by NVIDIA H100 Tensor Core GPUs, which feature fourth-generation Tensor Cores and NVLink interconnects, delivering significantly higher FP8, FP16, and BF16 throughput compared to prior GPU generations.
- SageMaker Studio manages the underlying EC2 instance lifecycle, allowing users to start, stop, and switch instances on demand, with compute costs accruing only when the instance is running.
- Workloads run within the familiar Jupyter or CodeEditor notebook environment, meaning existing training scripts, HuggingFace pipelines, PyTorch/TensorFlow code, and diffusion model libraries can be executed directly without modification.
- Users can access developer guides for JupyterLab and CodeEditor application setup to configure the correct kernel environments and IAM permissions needed to spin up P5.4xl instances within their Studio domain.
Why It's Important
- Bringing H100-class GPUs into the interactive notebook experience dramatically shortens the iteration cycle for researchers experimenting with large language models (LLMs) and diffusion models, eliminating the need to submit batch training jobs just to test ideas.
- The claimed 4x acceleration over previous-generation GPU instances (such as P4d/A100-based) means experiments that previously took hours can complete in a fraction of the time, directly improving researcher productivity.
- A reported 40% reduction in ML training costs makes it economically viable to run more experiments and larger models within budget constraints, lowering the barrier to generative AI development.
- Supporting use cases like question answering, code generation, image/video generation, and speech recognition in a single instance type simplifies infrastructure decisions for teams building multi-modal generative AI applications.
- Availability in SageMaker Studio means teams benefit from integrated MLOps tooling—experiment tracking, model registry, and data access—alongside the raw compute power of H100 GPUs.
How It's Different
- Unlike P4d instances (NVIDIA A100 GPUs), P5.4xl instances use NVIDIA H100 Tensor Core GPUs with fourth-generation Tensor Cores, delivering substantially higher throughput for transformer-based model training and inference.
- The P5.4xl is a mid-tier variant of the P5 family, offering a balance between cost and performance compared to the full p5.48xlarge (8x H100), making H100 compute accessible for teams that don't need the full 8-GPU configuration.
- Previous GPU instance availability in SageMaker Studio was largely limited to older generations (P3, P4); P5.4xl represents the first H100-based option natively available in the interactive Studio notebook environment.
- Compared to running P5 instances via SageMaker Training Jobs, the Studio notebook integration provides an interactive, low-latency development experience rather than a batch-oriented workflow, enabling real-time debugging and iterative experimentation.
- The H100's support for FP8 precision (in addition to BF16/FP16) enables more efficient training of large-scale models compared to A100-based instances, which lack native FP8 hardware support.
When to Prefer It
- Choose P5.4xl in Studio notebooks when interactively developing, fine-tuning, or debugging large language models (e.g., Llama, Mistral, Falcon) where rapid iteration and real-time feedback are critical.
- Prefer this instance type when training or running inference on diffusion models (e.g., Stable Diffusion, DALL-E style architectures) that require high GPU memory bandwidth and tensor compute throughput.
- Use P5.4xl when your workload benefits from H100-specific features such as FP8 precision training or high-speed NVLink memory bandwidth, and a single GPU partition is sufficient.
- This is the right choice for HPC workloads co-located with ML workflows, such as physics simulations or molecular dynamics, that require the raw floating-point performance of H100 hardware.
- Opt for P5.4xl over submitting SageMaker Training Jobs when you need an interactive environment for exploratory research, hyperparameter sweeps with manual inspection, or step-through debugging of GPU kernels.
- Consider P5.4xl when cost efficiency is a priority for H100 workloads—its 40% cost reduction claim over prior generations makes it preferable for sustained training runs that would otherwise be cost-prohibitive on older GPU types.
Availability
- Status: Generally Available (GA) as of May 11, 2026.
- Supported Regions: US East (N. Virginia), US East (Ohio), US West (Oregon), Asia Pacific (Mumbai), Asia Pacific (Tokyo), Asia Pacific (Jakarta), and South America (São Paulo).
- Supported Environments: Available within SageMaker Studio's JupyterLab and CodeEditor applications; not explicitly listed for SageMaker Classic notebooks.
- Pricing: On-demand pricing applies per the SageMaker Studio notebook instance pricing page; costs accrue only while the instance is active.
- Limitations: Not yet available in all AWS regions (notably absent from EU regions and additional Asia Pacific regions such as Singapore and Sydney at launch).
- Prerequisites: Users must have appropriate IAM permissions and SageMaker Studio domain configuration to select P5.4xl as a notebook instance type; service quota increases may be required for P5 instances in some accounts.