aws-sdk-sagemaker 1.318.0 → 1.320.0

data/lib/aws-sdk-sagemaker/types.rb

@@ -136,6 +136,27 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Specifies an instance group and the number of nodes to add to it.
+    #
+    # @!attribute [rw] instance_group_name
+    #   The name of the instance group to which you want to add nodes.
+    #   @return [String]
+    #
+    # @!attribute [rw] increment_target_count_by
+    #   The number of nodes to add to the specified instance group. The
+    #   total number of nodes across all instance groups in a single request
+    #   cannot exceed 50.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/AddClusterNodeSpecification AWS API Documentation
+    #
+    class AddClusterNodeSpecification < Struct.new(
+      :instance_group_name,
+      :increment_target_count_by)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] resource_arn
     #   The Amazon Resource Name (ARN) of the resource that you want to tag.
     #   @return [String]
@@ -172,6 +193,22 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Information about additional Elastic Network Interfaces (ENIs)
+    # associated with an instance.
+    #
+    # @!attribute [rw] efa_enis
+    #   A list of Elastic Fabric Adapter (EFA) ENIs associated with the
+    #   instance.
+    #   @return [Array<String>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/AdditionalEnis AWS API Documentation
+    #
+    class AdditionalEnis < Struct.new(
+      :efa_enis)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # A structure of additional Inference Specification. Additional
     # Inference Specification specifies details about inference jobs that
     # can be run with models based on this model package
@@ -3043,6 +3080,93 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Information about an error that occurred during the node addition
+    # operation.
+    #
+    # @!attribute [rw] instance_group_name
+    #   The name of the instance group for which the error occurred.
+    #   @return [String]
+    #
+    # @!attribute [rw] error_code
+    #   The error code associated with the failure. Possible values include
+    #   `InstanceGroupNotFound` and `InvalidInstanceGroupState`.
+    #   @return [String]
+    #
+    # @!attribute [rw] failed_count
+    #   The number of nodes that failed to be added to the specified
+    #   instance group.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] message
+    #   A descriptive message providing additional details about the error.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/BatchAddClusterNodesError AWS API Documentation
+    #
+    class BatchAddClusterNodesError < Struct.new(
+      :instance_group_name,
+      :error_code,
+      :failed_count,
+      :message)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] cluster_name
+    #   The name of the HyperPod cluster to which you want to add nodes.
+    #   @return [String]
+    #
+    # @!attribute [rw] client_token
+    #   A unique, case-sensitive identifier that you provide to ensure the
+    #   idempotency of the request. This token is valid for 8 hours. If you
+    #   retry the request with the same client token within this timeframe
+    #   and the same parameters, the API returns the same set of
+    #   `NodeLogicalIds` with their latest status.
+    #
+    #   **A suitable default value is auto-generated.** You should normally
+    #   not need to pass this option.
+    #   @return [String]
+    #
+    # @!attribute [rw] nodes_to_add
+    #   A list of instance groups and the number of nodes to add to each.
+    #   You can specify up to 5 instance groups in a single request, with a
+    #   maximum of 50 nodes total across all instance groups.
+    #   @return [Array<Types::AddClusterNodeSpecification>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/BatchAddClusterNodesRequest AWS API Documentation
+    #
+    class BatchAddClusterNodesRequest < Struct.new(
+      :cluster_name,
+      :client_token,
+      :nodes_to_add)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] successful
+    #   A list of `NodeLogicalIDs` that were successfully added to the
+    #   cluster. The `NodeLogicalID` is unique per cluster and does not
+    #   change between instance replacements. Each entry includes a
+    #   `NodeLogicalId` that can be used to track the node's provisioning
+    #   status (with `DescribeClusterNode`), the instance group name, and
+    #   the current status of the node.
+    #   @return [Array<Types::NodeAdditionResult>]
+    #
+    # @!attribute [rw] failed
+    #   A list of errors that occurred during the node addition operation.
+    #   Each entry includes the instance group name, error code, number of
+    #   failed additions, and an error message.
+    #   @return [Array<Types::BatchAddClusterNodesError>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/BatchAddClusterNodesResponse AWS API Documentation
+    #
+    class BatchAddClusterNodesResponse < Struct.new(
+      :successful,
+      :failed)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Configuration to control how SageMaker captures inference data for
     # batch transform jobs.
     #
@@ -3083,6 +3207,32 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Information about an error that occurred when attempting to delete a
+    # node identified by its `NodeLogicalId`.
+    #
+    # @!attribute [rw] code
+    #   The error code associated with the failure. Possible values include
+    #   `NodeLogicalIdNotFound`, `InvalidNodeStatus`, and `InternalError`.
+    #   @return [String]
+    #
+    # @!attribute [rw] message
+    #   A descriptive message providing additional details about the error.
+    #   @return [String]
+    #
+    # @!attribute [rw] node_logical_id
+    #   The `NodeLogicalId` of the node that could not be deleted.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/BatchDeleteClusterNodeLogicalIdsError AWS API Documentation
+    #
+    class BatchDeleteClusterNodeLogicalIdsError < Struct.new(
+      :code,
+      :message,
+      :node_logical_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Represents an error encountered when deleting a node from a SageMaker
     # HyperPod cluster.
     #
@@ -3136,11 +3286,19 @@ module Aws::SageMaker
     #   [1]: http://aws.amazon.com/contact-us/
     #   @return [Array<String>]
     #
+    # @!attribute [rw] node_logical_ids
+    #   A list of `NodeLogicalIds` identifying the nodes to be deleted. You
+    #   can specify up to 50 `NodeLogicalIds`. You must specify either
+    #   `NodeLogicalIds`, `InstanceIds`, or both, with a combined maximum of
+    #   50 identifiers.
+    #   @return [Array<String>]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/BatchDeleteClusterNodesRequest AWS API Documentation
     #
     class BatchDeleteClusterNodesRequest < Struct.new(
       :cluster_name,
-      :node_ids)
+      :node_ids,
+      :node_logical_ids)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -3154,11 +3312,23 @@ module Aws::SageMaker
     #   cluster.
     #   @return [Array<String>]
     #
+    # @!attribute [rw] failed_node_logical_ids
+    #   A list of `NodeLogicalIds` that could not be deleted, along with
+    #   error information explaining why the deletion failed.
+    #   @return [Array<Types::BatchDeleteClusterNodeLogicalIdsError>]
+    #
+    # @!attribute [rw] successful_node_logical_ids
+    #   A list of `NodeLogicalIds` that were successfully deleted from the
+    #   cluster.
+    #   @return [Array<String>]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/BatchDeleteClusterNodesResponse AWS API Documentation
     #
     class BatchDeleteClusterNodesResponse < Struct.new(
       :failed,
-      :successful)
+      :successful,
+      :failed_node_logical_ids,
+      :successful_node_logical_ids)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -3633,6 +3803,27 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Information about the Capacity Reservation used by an instance or
+    # instance group.
+    #
+    # @!attribute [rw] arn
+    #   The Amazon Resource Name (ARN) of the Capacity Reservation.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The type of Capacity Reservation. Valid values are `ODCR` (On-Demand
+    #   Capacity Reservation) or `CRG` (Capacity Reservation Group).
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CapacityReservation AWS API Documentation
+    #
+    class CapacityReservation < Struct.new(
+      :arn,
+      :type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Specifies the type and size of the endpoint capacity to activate for a
     # blue/green deployment, a rolling deployment, or a rollback strategy.
     # You can specify your batches as either instance count or the overall
@@ -4534,6 +4725,117 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Detailed information about a specific event in a HyperPod cluster.
+    #
+    # @!attribute [rw] event_id
+    #   The unique identifier (UUID) of the event.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   The Amazon Resource Name (ARN) of the HyperPod cluster associated
+    #   with the event.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_name
+    #   The name of the HyperPod cluster associated with the event.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_group_name
+    #   The name of the instance group associated with the event, if
+    #   applicable.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_id
+    #   The EC2 instance ID associated with the event, if applicable.
+    #   @return [String]
+    #
+    # @!attribute [rw] resource_type
+    #   The type of resource associated with the event. Valid values are
+    #   `Cluster`, `InstanceGroup`, or `Instance`.
+    #   @return [String]
+    #
+    # @!attribute [rw] event_time
+    #   The timestamp when the event occurred.
+    #   @return [Time]
+    #
+    # @!attribute [rw] event_details
+    #   Additional details about the event, including event-specific
+    #   metadata.
+    #   @return [Types::EventDetails]
+    #
+    # @!attribute [rw] description
+    #   A human-readable description of the event.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterEventDetail AWS API Documentation
+    #
+    class ClusterEventDetail < Struct.new(
+      :event_id,
+      :cluster_arn,
+      :cluster_name,
+      :instance_group_name,
+      :instance_id,
+      :resource_type,
+      :event_time,
+      :event_details,
+      :description)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A summary of an event in a HyperPod cluster.
+    #
+    # @!attribute [rw] event_id
+    #   The unique identifier (UUID) of the event.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   The Amazon Resource Name (ARN) of the HyperPod cluster associated
+    #   with the event.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_name
+    #   The name of the HyperPod cluster associated with the event.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_group_name
+    #   The name of the instance group associated with the event, if
+    #   applicable.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_id
+    #   The Amazon Elastic Compute Cloud (EC2) instance ID associated with
+    #   the event, if applicable.
+    #   @return [String]
+    #
+    # @!attribute [rw] resource_type
+    #   The type of resource associated with the event. Valid values are
+    #   `Cluster`, `InstanceGroup`, or `Instance`.
+    #   @return [String]
+    #
+    # @!attribute [rw] event_time
+    #   The timestamp when the event occurred.
+    #   @return [Time]
+    #
+    # @!attribute [rw] description
+    #   A brief, human-readable description of the event.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterEventSummary AWS API Documentation
+    #
+    class ClusterEventSummary < Struct.new(
+      :event_id,
+      :cluster_arn,
+      :cluster_name,
+      :instance_group_name,
+      :instance_id,
+      :resource_type,
+      :event_time,
+      :description)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Details of an instance group in a SageMaker HyperPod cluster.
     #
     # @!attribute [rw] current_count
@@ -4630,6 +4932,16 @@ module Aws::SageMaker
     #   updating the AMI.
     #   @return [Types::ScheduledUpdateConfig]
     #
+    # @!attribute [rw] current_image_id
+    #   The ID of the Amazon Machine Image (AMI) currently in use by the
+    #   instance group.
+    #   @return [String]
+    #
+    # @!attribute [rw] desired_image_id
+    #   The ID of the Amazon Machine Image (AMI) desired for the instance
+    #   group.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterInstanceGroupDetails AWS API Documentation
     #
     class ClusterInstanceGroupDetails < Struct.new(
@@ -4646,7 +4958,9 @@ module Aws::SageMaker
       :training_plan_arn,
       :training_plan_status,
       :override_vpc_config,
-      :scheduled_update_config)
+      :scheduled_update_config,
+      :current_image_id,
+      :desired_image_id)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -4756,6 +5070,33 @@ module Aws::SageMaker
     #   update the AMI.
     #   @return [Types::ScheduledUpdateConfig]
     #
+    # @!attribute [rw] image_id
+    #   When configuring your HyperPod cluster, you can specify an image ID
+    #   using one of the following options:
+    #
+    #   * `HyperPodPublicAmiId`: Use a HyperPod public AMI
+    #
+    #   * `CustomAmiId`: Use your custom AMI
+    #
+    #   * `default`: Use the default latest system image
+    #
+    #   f you choose to use a custom AMI (`CustomAmiId`), ensure it meets
+    #   the following requirements:
+    #
+    #   * Encryption: The custom AMI must be unencrypted.
+    #
+    #   * Ownership: The custom AMI must be owned by the same Amazon Web
+    #     Services account that is creating the HyperPod cluster.
+    #
+    #   * Volume support: Only the primary AMI snapshot volume is supported;
+    #     additional AMI volumes are not supported.
+    #
+    #   When updating the instance group's AMI through the
+    #   `UpdateClusterSoftware` operation, if an instance group uses a
+    #   custom AMI, you must provide an `ImageId` or use the default as
+    #   input.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterInstanceGroupSpecification AWS API Documentation
     #
     class ClusterInstanceGroupSpecification < Struct.new(
@@ -4769,7 +5110,8 @@ module Aws::SageMaker
       :on_start_deep_health_checks,
       :training_plan_arn,
       :override_vpc_config,
-      :scheduled_update_config)
+      :scheduled_update_config,
+      :image_id)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -4879,6 +5221,37 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Metadata information about a HyperPod cluster showing information
+    # about the cluster level operations, such as creating, updating, and
+    # deleting.
+    #
+    # @!attribute [rw] failure_message
+    #   An error message describing why the cluster level operation (such as
+    #   creating, updating, or deleting) failed.
+    #   @return [String]
+    #
+    # @!attribute [rw] eks_role_access_entries
+    #   A list of Amazon EKS IAM role ARNs associated with the cluster. This
+    #   is created by HyperPod on your behalf and only applies for EKS
+    #   orchestrated clusters.
+    #   @return [Array<String>]
+    #
+    # @!attribute [rw] slr_access_entry
+    #   The Service-Linked Role (SLR) associated with the cluster. This is
+    #   created by HyperPod on your behalf and only applies for EKS
+    #   orchestrated clusters.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterMetadata AWS API Documentation
+    #
+    class ClusterMetadata < Struct.new(
+      :failure_message,
+      :eks_role_access_entries,
+      :slr_access_entry)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Details of an instance (also called a *node* interchangeably) in a
     # SageMaker HyperPod cluster.
     #
@@ -4890,6 +5263,13 @@ module Aws::SageMaker
     #   The ID of the instance.
     #   @return [String]
     #
+    # @!attribute [rw] node_logical_id
+    #   A unique identifier for the node that persists throughout its
+    #   lifecycle, from provisioning request to termination. This identifier
+    #   can be used to track the node even before it has an assigned
+    #   `InstanceId`.
+    #   @return [String]
+    #
     # @!attribute [rw] instance_status
     #   The status of the instance.
     #   @return [Types::ClusterInstanceStatusDetails]
@@ -4947,11 +5327,25 @@ module Aws::SageMaker
     #   The placement details of the SageMaker HyperPod cluster node.
     #   @return [Types::ClusterInstancePlacement]
     #
+    # @!attribute [rw] current_image_id
+    #   The ID of the Amazon Machine Image (AMI) currently in use by the
+    #   node.
+    #   @return [String]
+    #
+    # @!attribute [rw] desired_image_id
+    #   The ID of the Amazon Machine Image (AMI) desired for the node.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_info
+    #   Contains information about the UltraServer.
+    #   @return [Types::UltraServerInfo]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterNodeDetails AWS API Documentation
     #
     class ClusterNodeDetails < Struct.new(
       :instance_group_name,
       :instance_id,
+      :node_logical_id,
       :instance_status,
       :instance_type,
       :launch_time,
@@ -4963,7 +5357,10 @@ module Aws::SageMaker
       :private_primary_ip,
       :private_primary_ipv_6,
       :private_dns_hostname,
-      :placement)
+      :placement,
+      :current_image_id,
+      :desired_image_id,
+      :ultra_server_info)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -4979,6 +5376,15 @@ module Aws::SageMaker
     #   The ID of the instance.
     #   @return [String]
     #
+    # @!attribute [rw] node_logical_id
+    #   A unique identifier for the node that persists throughout its
+    #   lifecycle, from provisioning request to termination. This identifier
+    #   can be used to track the node even before it has an assigned
+    #   `InstanceId`. This field is only included when
+    #   `IncludeNodeLogicalIds` is set to `True` in the `ListClusterNodes`
+    #   request.
+    #   @return [String]
+    #
     # @!attribute [rw] instance_type
     #   The type of the instance.
     #   @return [String]
@@ -4996,15 +5402,21 @@ module Aws::SageMaker
     #   The status of the instance.
     #   @return [Types::ClusterInstanceStatusDetails]
     #
+    # @!attribute [rw] ultra_server_info
+    #   Contains information about the UltraServer.
+    #   @return [Types::UltraServerInfo]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterNodeSummary AWS API Documentation
     #
     class ClusterNodeSummary < Struct.new(
       :instance_group_name,
       :instance_id,
+      :node_logical_id,
       :instance_type,
       :launch_time,
       :last_software_update_time,
-      :instance_status)
+      :instance_status,
+      :ultra_server_info)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -6822,7 +7234,7 @@ module Aws::SageMaker
     # @!attribute [rw] orchestrator
     #   The type of orchestrator to use for the SageMaker HyperPod cluster.
     #   Currently, the only supported value is `"eks"`, which is to use an
-    #   Amazon Elastic Kubernetes Service (EKS) cluster as the orchestrator.
+    #   Amazon Elastic Kubernetes Service cluster as the orchestrator.
     #   @return [Types::ClusterOrchestrator]
     #
     # @!attribute [rw] node_recovery
@@ -6833,6 +7245,21 @@ module Aws::SageMaker
     #   cluster instances.
     #   @return [String]
     #
+    # @!attribute [rw] node_provisioning_mode
+    #   The mode for provisioning nodes in the cluster. You can specify the
+    #   following modes:
+    #
+    #   * **Continuous**: Scaling behavior that enables 1) concurrent
+    #     operation execution within instance groups, 2) continuous retry
+    #     mechanisms for failed operations, 3) enhanced customer visibility
+    #     into cluster events through detailed event streams, 4) partial
+    #     provisioning capabilities. Your clusters and instance groups
+    #     remain `InService` while scaling. This mode is only supported for
+    #     EKS orchestrated clusters.
+    #
+    #   ^
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateClusterRequest AWS API Documentation
     #
     class CreateClusterRequest < Struct.new(
@@ -6842,7 +7269,8 @@ module Aws::SageMaker
       :vpc_config,
       :tags,
       :orchestrator,
-      :node_recovery)
+      :node_recovery,
+      :node_provisioning_mode)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -11200,6 +11628,11 @@ module Aws::SageMaker
     #   creating this plan.
     #   @return [String]
     #
+    # @!attribute [rw] spare_instance_count_per_ultra_server
+    #   Number of spare instances to reserve per UltraServer for enhanced
+    #   resiliency. Default is 1.
+    #   @return [Integer]
+    #
     # @!attribute [rw] tags
     #   An array of key-value pairs to apply to this training plan.
     #   @return [Array<Types::Tag>]
@@ -11209,6 +11642,7 @@ module Aws::SageMaker
     class CreateTrainingPlanRequest < Struct.new(
       :training_plan_name,
       :training_plan_offering_id,
+      :spare_instance_count_per_ultra_server,
       :tags)
       SENSITIVE = []
       include Aws::Structure
@@ -14391,6 +14825,39 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] event_id
+    #   The unique identifier (UUID) of the event to describe. This ID can
+    #   be obtained from the `ListClusterEvents` operation.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_name
+    #   The name or Amazon Resource Name (ARN) of the HyperPod cluster
+    #   associated with the event.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterEventRequest AWS API Documentation
+    #
+    class DescribeClusterEventRequest < Struct.new(
+      :event_id,
+      :cluster_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] event_details
+    #   Detailed information about the requested cluster event, including
+    #   event metadata for various resource types such as `Cluster`,
+    #   `InstanceGroup`, `Instance`, and their associated attributes.
+    #   @return [Types::ClusterEventDetail]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterEventResponse AWS API Documentation
+    #
+    class DescribeClusterEventResponse < Struct.new(
+      :event_details)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] cluster_name
     #   The string name or the Amazon Resource Name (ARN) of the SageMaker
     #   HyperPod cluster in which the node is.
@@ -14400,11 +14867,19 @@ module Aws::SageMaker
     #   The ID of the SageMaker HyperPod cluster node.
     #   @return [String]
     #
+    # @!attribute [rw] node_logical_id
+    #   The logical identifier of the node to describe. You can specify
+    #   either `NodeLogicalId` or `InstanceId`, but not both.
+    #   `NodeLogicalId` can be used to describe nodes that are still being
+    #   provisioned and don't yet have an `InstanceId` assigned.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterNodeRequest AWS API Documentation
     #
     class DescribeClusterNodeRequest < Struct.new(
       :cluster_name,
-      :node_id)
+      :node_id,
+      :node_logical_id)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -14484,6 +14959,10 @@ module Aws::SageMaker
     #   cluster.
     #   @return [String]
     #
+    # @!attribute [rw] node_provisioning_mode
+    #   The mode used for provisioning nodes in the cluster.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterResponse AWS API Documentation
     #
     class DescribeClusterResponse < Struct.new(
@@ -14496,7 +14975,8 @@ module Aws::SageMaker
       :restricted_instance_groups,
       :vpc_config,
       :orchestrator,
-      :node_recovery)
+      :node_recovery,
+      :node_provisioning_mode)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -19216,6 +19696,94 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] reserved_capacity_arn
+    #   ARN of the reserved capacity to describe.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeReservedCapacityRequest AWS API Documentation
+    #
+    class DescribeReservedCapacityRequest < Struct.new(
+      :reserved_capacity_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] reserved_capacity_arn
+    #   ARN of the reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] reserved_capacity_type
+    #   The type of reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The current status of the reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] availability_zone
+    #   The Availability Zone where the reserved capacity is provisioned.
+    #   @return [String]
+    #
+    # @!attribute [rw] duration_hours
+    #   The total duration of the reserved capacity in hours.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] duration_minutes
+    #   The number of minutes for the duration of the reserved capacity. For
+    #   example, if a reserved capacity starts at 08:55 and ends at 11:30,
+    #   the minutes field would be 35.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] start_time
+    #   The timestamp when the reserved capacity becomes active.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time
+    #   The timestamp when the reserved capacity expires.
+    #   @return [Time]
+    #
+    # @!attribute [rw] instance_type
+    #   The Amazon EC2 instance type used in the reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] total_instance_count
+    #   The total number of instances allocated to this reserved capacity.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] available_instance_count
+    #   The number of instances currently available for use in this reserved
+    #   capacity.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] in_use_instance_count
+    #   The number of instances currently in use from this reserved
+    #   capacity.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] ultra_server_summary
+    #   A summary of the UltraServer associated with this reserved capacity.
+    #   @return [Types::UltraServerSummary]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeReservedCapacityResponse AWS API Documentation
+    #
+    class DescribeReservedCapacityResponse < Struct.new(
+      :reserved_capacity_arn,
+      :reserved_capacity_type,
+      :status,
+      :availability_zone,
+      :duration_hours,
+      :duration_minutes,
+      :start_time,
+      :end_time,
+      :instance_type,
+      :total_instance_count,
+      :available_instance_count,
+      :in_use_instance_count,
+      :ultra_server_summary)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] domain_id
     #   The ID of the associated domain.
     #   @return [String]
@@ -19883,6 +20451,19 @@ module Aws::SageMaker
     #   The number of instances currently in use from this training plan.
     #   @return [Integer]
     #
+    # @!attribute [rw] unhealthy_instance_count
+    #   The number of instances in the training plan that are currently in
+    #   an unhealthy state.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] available_spare_instance_count
+    #   The number of available spare instances in the training plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] total_ultra_server_count
+    #   The total number of UltraServers reserved to this training plan.
+    #   @return [Integer]
+    #
     # @!attribute [rw] target_resources
     #   The target resources (e.g., SageMaker Training Jobs, SageMaker
     #   HyperPod) that can use this training plan.
@@ -19917,6 +20498,9 @@ module Aws::SageMaker
       :total_instance_count,
       :available_instance_count,
       :in_use_instance_count,
+      :unhealthy_instance_count,
+      :available_spare_instance_count,
+      :total_ultra_server_count,
       :target_resources,
       :reserved_capacity_summaries)
       SENSITIVE = []
@@ -22293,6 +22877,61 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Detailed information about a specific event, including event metadata.
+    #
+    # @!attribute [rw] event_metadata
+    #   Metadata specific to the event, which may include information about
+    #   the cluster, instance group, or instance involved.
+    #   @return [Types::EventMetadata]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/EventDetails AWS API Documentation
+    #
+    class EventDetails < Struct.new(
+      :event_metadata)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Metadata associated with a cluster event, which may include details
+    # about various resource types.
+    #
+    # @note EventMetadata is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of EventMetadata corresponding to the set member.
+    #
+    # @!attribute [rw] cluster
+    #   Metadata specific to cluster-level events.
+    #   @return [Types::ClusterMetadata]
+    #
+    # @!attribute [rw] instance_group
+    #   Metadata specific to instance group-level events.
+    #   @return [Types::InstanceGroupMetadata]
+    #
+    # @!attribute [rw] instance_group_scaling
+    #   Metadata related to instance group scaling events.
+    #   @return [Types::InstanceGroupScalingMetadata]
+    #
+    # @!attribute [rw] instance
+    #   Metadata specific to instance-level events.
+    #   @return [Types::InstanceMetadata]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/EventMetadata AWS API Documentation
+    #
+    class EventMetadata < Struct.new(
+      :cluster,
+      :instance_group,
+      :instance_group_scaling,
+      :instance,
+      :unknown)
+      SENSITIVE = []
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class Cluster < EventMetadata; end
+      class InstanceGroup < EventMetadata; end
+      class InstanceGroupScaling < EventMetadata; end
+      class Instance < EventMetadata; end
+      class Unknown < EventMetadata; end
+    end
+
     # The properties of an experiment as returned by the [Search][1] API.
     # For information about experiments, see the [CreateExperiment][2] API.
     #
@@ -27531,6 +28170,116 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Metadata information about an instance group in a HyperPod cluster.
+    #
+    # @!attribute [rw] failure_message
+    #   An error message describing why the instance group level operation
+    #   (such as creating, scaling, or deleting) failed.
+    #   @return [String]
+    #
+    # @!attribute [rw] availability_zone_id
+    #   The ID of the Availability Zone where the instance group is located.
+    #   @return [String]
+    #
+    # @!attribute [rw] capacity_reservation
+    #   Information about the Capacity Reservation used by the instance
+    #   group.
+    #   @return [Types::CapacityReservation]
+    #
+    # @!attribute [rw] subnet_id
+    #   The ID of the subnet where the instance group is located.
+    #   @return [String]
+    #
+    # @!attribute [rw] security_group_ids
+    #   A list of security group IDs associated with the instance group.
+    #   @return [Array<String>]
+    #
+    # @!attribute [rw] ami_override
+    #   If you use a custom Amazon Machine Image (AMI) for the instance
+    #   group, this field shows the ID of the custom AMI.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/InstanceGroupMetadata AWS API Documentation
+    #
+    class InstanceGroupMetadata < Struct.new(
+      :failure_message,
+      :availability_zone_id,
+      :capacity_reservation,
+      :subnet_id,
+      :security_group_ids,
+      :ami_override)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Metadata information about scaling operations for an instance group.
+    #
+    # @!attribute [rw] instance_count
+    #   The current number of instances in the group.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] target_count
+    #   The desired number of instances for the group after scaling.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] failure_message
+    #   An error message describing why the scaling operation failed, if
+    #   applicable.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/InstanceGroupScalingMetadata AWS API Documentation
+    #
+    class InstanceGroupScalingMetadata < Struct.new(
+      :instance_count,
+      :target_count,
+      :failure_message)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Metadata information about an instance in a HyperPod cluster.
+    #
+    # @!attribute [rw] customer_eni
+    #   The ID of the customer-managed Elastic Network Interface (ENI)
+    #   associated with the instance.
+    #   @return [String]
+    #
+    # @!attribute [rw] additional_enis
+    #   Information about additional Elastic Network Interfaces (ENIs)
+    #   associated with the instance.
+    #   @return [Types::AdditionalEnis]
+    #
+    # @!attribute [rw] capacity_reservation
+    #   Information about the Capacity Reservation used by the instance.
+    #   @return [Types::CapacityReservation]
+    #
+    # @!attribute [rw] failure_message
+    #   An error message describing why the instance creation or update
+    #   failed, if applicable.
+    #   @return [String]
+    #
+    # @!attribute [rw] lcs_execution_state
+    #   The execution state of the Lifecycle Script (LCS) for the instance.
+    #   @return [String]
+    #
+    # @!attribute [rw] node_logical_id
+    #   The unique logical identifier of the node within the cluster. The ID
+    #   used here is the same object as in the `BatchAddClusterNodes` API.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/InstanceMetadata AWS API Documentation
+    #
+    class InstanceMetadata < Struct.new(
+      :customer_eni,
+      :additional_enis,
+      :capacity_reservation,
+      :failure_message,
+      :lcs_execution_state,
+      :node_logical_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Information on the IMDS configuration of the notebook instance
     #
     # @!attribute [rw] minimum_instance_metadata_service_version
@@ -27549,6 +28298,30 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Configuration for how instances are placed and allocated within
+    # UltraServers. This is only applicable for UltraServer capacity.
+    #
+    # @!attribute [rw] enable_multiple_jobs
+    #   If set to true, allows multiple jobs to share the same UltraServer
+    #   instances. If set to false, ensures this job's instances are placed
+    #   on an UltraServer exclusively, with no other jobs sharing the same
+    #   UltraServer. Default is false.
+    #   @return [Boolean]
+    #
+    # @!attribute [rw] placement_specifications
+    #   A list of specifications for how instances should be placed on
+    #   specific UltraServers. Maximum of 10 items is supported.
+    #   @return [Array<Types::PlacementSpecification>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/InstancePlacementConfig AWS API Documentation
+    #
+    class InstancePlacementConfig < Struct.new(
+      :enable_multiple_jobs,
+      :placement_specifications)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # For a hyperparameter of the integer type, specifies the range that a
     # hyperparameter tuning job searches.
     #
@@ -29072,6 +29845,91 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] cluster_name
+    #   The name or Amazon Resource Name (ARN) of the HyperPod cluster for
+    #   which to list events.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_group_name
+    #   The name of the instance group to filter events. If specified, only
+    #   events related to this instance group are returned.
+    #   @return [String]
+    #
+    # @!attribute [rw] node_id
+    #   The EC2 instance ID to filter events. If specified, only events
+    #   related to this instance are returned.
+    #   @return [String]
+    #
+    # @!attribute [rw] event_time_after
+    #   The start of the time range for filtering events. Only events that
+    #   occurred after this time are included in the results.
+    #   @return [Time]
+    #
+    # @!attribute [rw] event_time_before
+    #   The end of the time range for filtering events. Only events that
+    #   occurred before this time are included in the results.
+    #   @return [Time]
+    #
+    # @!attribute [rw] sort_by
+    #   The field to use for sorting the event list. Currently, the only
+    #   supported value is `EventTime`.
+    #   @return [String]
+    #
+    # @!attribute [rw] sort_order
+    #   The order in which to sort the results. Valid values are `Ascending`
+    #   or `Descending` (the default is `Descending`).
+    #   @return [String]
+    #
+    # @!attribute [rw] resource_type
+    #   The type of resource for which to filter events. Valid values are
+    #   `Cluster`, `InstanceGroup`, or `Instance`.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   The maximum number of events to return in the response. Valid range
+    #   is 1 to 100.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] next_token
+    #   A token to retrieve the next set of results. This token is obtained
+    #   from the output of a previous `ListClusterEvents` call.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListClusterEventsRequest AWS API Documentation
+    #
+    class ListClusterEventsRequest < Struct.new(
+      :cluster_name,
+      :instance_group_name,
+      :node_id,
+      :event_time_after,
+      :event_time_before,
+      :sort_by,
+      :sort_order,
+      :resource_type,
+      :max_results,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] next_token
+    #   A token to retrieve the next set of results. Include this token in
+    #   subsequent `ListClusterEvents` calls to fetch more events.
+    #   @return [String]
+    #
+    # @!attribute [rw] events
+    #   A list of event summaries matching the specified criteria.
+    #   @return [Array<Types::ClusterEventSummary>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListClusterEventsResponse AWS API Documentation
+    #
+    class ListClusterEventsResponse < Struct.new(
+      :next_token,
+      :events)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] cluster_name
     #   The string name or the Amazon Resource Name (ARN) of the SageMaker
     #   HyperPod cluster in which you want to retrieve the list of nodes.
@@ -29140,6 +29998,13 @@ module Aws::SageMaker
     #   The sort order for results. The default value is `Ascending`.
     #   @return [String]
     #
+    # @!attribute [rw] include_node_logical_ids
+    #   Specifies whether to include nodes that are still being provisioned
+    #   in the response. When set to true, the response includes all nodes
+    #   regardless of their provisioning status. When set to `False`
+    #   (default), only nodes with assigned `InstanceIds` are returned.
+    #   @return [Boolean]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListClusterNodesRequest AWS API Documentation
     #
     class ListClusterNodesRequest < Struct.new(
@@ -29150,7 +30015,8 @@ module Aws::SageMaker
       :max_results,
       :next_token,
       :sort_by,
-      :sort_order)
+      :sort_order,
+      :include_node_logical_ids)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -34283,6 +35149,49 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] reserved_capacity_arn
+    #   The ARN of the reserved capacity to list UltraServers for.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   The maximum number of UltraServers to return in the response. The
+    #   default value is 10.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] next_token
+    #   If the previous response was truncated, you receive this token. Use
+    #   it in your next request to receive the next set of results.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListUltraServersByReservedCapacityRequest AWS API Documentation
+    #
+    class ListUltraServersByReservedCapacityRequest < Struct.new(
+      :reserved_capacity_arn,
+      :max_results,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] next_token
+    #   If the response is truncated, SageMaker returns this token. Use it
+    #   in the next request to retrieve the next set of UltraServers.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_servers
+    #   A list of UltraServers that are part of the specified reserved
+    #   capacity.
+    #   @return [Array<Types::UltraServer>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListUltraServersByReservedCapacityResponse AWS API Documentation
+    #
+    class ListUltraServersByReservedCapacityResponse < Struct.new(
+      :next_token,
+      :ultra_servers)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] next_token
     #   If the previous response was truncated, you will receive this token.
     #   Use it in your next request to receive the next set of results.
@@ -37695,6 +38604,33 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Information about a node that was successfully added to the cluster.
+    #
+    # @!attribute [rw] node_logical_id
+    #   A unique identifier assigned to the node that can be used to track
+    #   its provisioning status through the `DescribeClusterNode` operation.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_group_name
+    #   The name of the instance group to which the node was added.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The current status of the node. Possible values include `Pending`,
+    #   `Running`, `Failed`, `ShuttingDown`, `SystemUpdating`,
+    #   `DeepHealthCheckInProgress`, and `NotFound`.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/NodeAdditionResult AWS API Documentation
+    #
+    class NodeAdditionResult < Struct.new(
+      :node_logical_id,
+      :instance_group_name,
+      :status)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Provides a summary of a notebook instance lifecycle configuration.
     #
     # @!attribute [rw] notebook_instance_lifecycle_config_name
@@ -39813,6 +40749,27 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Specifies how instances should be placed on a specific UltraServer.
+    #
+    # @!attribute [rw] ultra_server_id
+    #   The unique identifier of the UltraServer where instances should be
+    #   placed.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_count
+    #   The number of ML compute instances required to be placed together on
+    #   the same UltraServer. Minimum value of 1.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/PlacementSpecification AWS API Documentation
+    #
+    class PlacementSpecification < Struct.new(
+      :ultra_server_id,
+      :instance_count)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # A specification for a predefined metric.
     #
     # @!attribute [rw] predefined_metric_type
@@ -42691,6 +43648,20 @@ module Aws::SageMaker
     # SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see
     # ` CreateTrainingPlan `.
     #
+    # @!attribute [rw] reserved_capacity_type
+    #   The type of reserved capacity offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_type
+    #   The type of UltraServer included in this reserved capacity offering,
+    #   such as ml.u-p6e-gb200x72.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_count
+    #   The number of UltraServers included in this reserved capacity
+    #   offering.
+    #   @return [Integer]
+    #
     # @!attribute [rw] instance_type
     #   The instance type for the reserved capacity offering.
     #   @return [String]
@@ -42724,6 +43695,9 @@ module Aws::SageMaker
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ReservedCapacityOffering AWS API Documentation
     #
     class ReservedCapacityOffering < Struct.new(
+      :reserved_capacity_type,
+      :ultra_server_type,
+      :ultra_server_count,
       :instance_type,
       :instance_count,
       :availability_zone,
@@ -42745,6 +43719,19 @@ module Aws::SageMaker
     #   The Amazon Resource Name (ARN); of the reserved capacity.
     #   @return [String]
     #
+    # @!attribute [rw] reserved_capacity_type
+    #   The type of reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_type
+    #   The type of UltraServer included in this reserved capacity, such as
+    #   ml.u-p6e-gb200x72.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_count
+    #   The number of UltraServers included in this reserved capacity.
+    #   @return [Integer]
+    #
     # @!attribute [rw] instance_type
     #   The instance type for the reserved capacity.
     #   @return [String]
@@ -42783,6 +43770,9 @@ module Aws::SageMaker
     #
     class ReservedCapacitySummary < Struct.new(
       :reserved_capacity_arn,
+      :reserved_capacity_type,
+      :ultra_server_type,
+      :ultra_server_count,
       :instance_type,
       :total_instance_count,
       :status,
@@ -42955,6 +43945,12 @@ module Aws::SageMaker
     #   resource configuration.
     #   @return [String]
     #
+    # @!attribute [rw] instance_placement_config
+    #   Configuration for how training job instances are placed and
+    #   allocated within UltraServers. Only applicable for UltraServer
+    #   capacity.
+    #   @return [Types::InstancePlacementConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ResourceConfig AWS API Documentation
     #
     class ResourceConfig < Struct.new(
@@ -42964,7 +43960,8 @@ module Aws::SageMaker
       :volume_kms_key_id,
       :keep_alive_period_in_seconds,
       :instance_groups,
-      :training_plan_arn)
+      :training_plan_arn,
+      :instance_placement_config)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -44140,6 +45137,14 @@ module Aws::SageMaker
     #   match your requirements.
     #   @return [Integer]
     #
+    # @!attribute [rw] ultra_server_type
+    #   The type of UltraServer to search for, such as ml.u-p6e-gb200x72.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_count
+    #   The number of UltraServers to search for.
+    #   @return [Integer]
+    #
     # @!attribute [rw] start_time_after
     #   A filter to search for training plan offerings with a start time
     #   after a specified date.
@@ -44172,6 +45177,8 @@ module Aws::SageMaker
     class SearchTrainingPlanOfferingsRequest < Struct.new(
       :instance_type,
       :instance_count,
+      :ultra_server_type,
+      :ultra_server_count,
       :start_time_after,
       :end_time_before,
       :duration_hours,
@@ -47523,6 +48530,10 @@ module Aws::SageMaker
     #   The number of instances currently in use from this training plan.
     #   @return [Integer]
     #
+    # @!attribute [rw] total_ultra_server_count
+    #   The total number of UltraServers allocated to this training plan.
+    #   @return [Integer]
+    #
     # @!attribute [rw] target_resources
     #   The target resources (e.g., training jobs, HyperPod clusters) that
     #   can use this training plan.
@@ -47558,6 +48569,7 @@ module Aws::SageMaker
       :total_instance_count,
       :available_instance_count,
       :in_use_instance_count,
+      :total_ultra_server_count,
       :target_resources,
       :reserved_capacity_summaries)
       SENSITIVE = []
@@ -49072,6 +50084,125 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Represents a high-performance compute server used for distributed
+    # training in SageMaker AI. An UltraServer consists of multiple
+    # instances within a shared NVLink interconnect domain.
+    #
+    # @!attribute [rw] ultra_server_id
+    #   The unique identifier for the UltraServer.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_type
+    #   The type of UltraServer, such as ml.u-p6e-gb200x72.
+    #   @return [String]
+    #
+    # @!attribute [rw] availability_zone
+    #   The name of the Availability Zone where the UltraServer is
+    #   provisioned.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_type
+    #   The Amazon EC2 instance type used in the UltraServer.
+    #   @return [String]
+    #
+    # @!attribute [rw] total_instance_count
+    #   The total number of instances in this UltraServer.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] configured_spare_instance_count
+    #   The number of spare instances configured for this UltraServer to
+    #   provide enhanced resiliency.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] available_instance_count
+    #   The number of instances currently available for use in this
+    #   UltraServer.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] in_use_instance_count
+    #   The number of instances currently in use in this UltraServer.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] available_spare_instance_count
+    #   The number of available spare instances in the UltraServer.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] unhealthy_instance_count
+    #   The number of instances in this UltraServer that are currently in an
+    #   unhealthy state.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] health_status
+    #   The overall health status of the UltraServer.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UltraServer AWS API Documentation
+    #
+    class UltraServer < Struct.new(
+      :ultra_server_id,
+      :ultra_server_type,
+      :availability_zone,
+      :instance_type,
+      :total_instance_count,
+      :configured_spare_instance_count,
+      :available_instance_count,
+      :in_use_instance_count,
+      :available_spare_instance_count,
+      :unhealthy_instance_count,
+      :health_status)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains information about the UltraServer object.
+    #
+    # @!attribute [rw] id
+    #   The unique identifier of the UltraServer.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UltraServerInfo AWS API Documentation
+    #
+    class UltraServerInfo < Struct.new(
+      :id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A summary of UltraServer resources and their current status.
+    #
+    # @!attribute [rw] ultra_server_type
+    #   The type of UltraServer, such as ml.u-p6e-gb200x72.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_type
+    #   The Amazon EC2 instance type used in the UltraServer.
+    #   @return [String]
+    #
+    # @!attribute [rw] ultra_server_count
+    #   The number of UltraServers of this type.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] available_spare_instance_count
+    #   The number of available spare instances in the UltraServers.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] unhealthy_instance_count
+    #   The total number of instances across all UltraServers of this type
+    #   that are currently in an unhealthy state.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UltraServerSummary AWS API Documentation
+    #
+    class UltraServerSummary < Struct.new(
+      :ultra_server_type,
+      :instance_type,
+      :ultra_server_count,
+      :available_spare_instance_count,
+      :unhealthy_instance_count)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The settings that apply to an Amazon SageMaker AI domain when you use
     # it in Amazon SageMaker Unified Studio.
     #
@@ -49393,12 +50524,40 @@ module Aws::SageMaker
     #   The configuration to use when updating the AMI versions.
     #   @return [Types::DeploymentConfiguration]
     #
+    # @!attribute [rw] image_id
+    #   When configuring your HyperPod cluster, you can specify an image ID
+    #   using one of the following options:
+    #
+    #   * `HyperPodPublicAmiId`: Use a HyperPod public AMI
+    #
+    #   * `CustomAmiId`: Use your custom AMI
+    #
+    #   * `default`: Use the default latest system image
+    #
+    #   f you choose to use a custom AMI (`CustomAmiId`), ensure it meets
+    #   the following requirements:
+    #
+    #   * Encryption: The custom AMI must be unencrypted.
+    #
+    #   * Ownership: The custom AMI must be owned by the same Amazon Web
+    #     Services account that is creating the HyperPod cluster.
+    #
+    #   * Volume support: Only the primary AMI snapshot volume is supported;
+    #     additional AMI volumes are not supported.
+    #
+    #   When updating the instance group's AMI through the
+    #   `UpdateClusterSoftware` operation, if an instance group uses a
+    #   custom AMI, you must provide an `ImageId` or use the default as
+    #   input.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdateClusterSoftwareRequest AWS API Documentation
     #
     class UpdateClusterSoftwareRequest < Struct.new(
       :cluster_name,
       :instance_groups,
-      :deployment_config)
+      :deployment_config,
+      :image_id)
       SENSITIVE = []
       include Aws::Structure
     end