aws-sdk-sagemaker 1.340.0 → 1.341.0

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

@@ -5145,6 +5145,26 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Defines the instance capacity requirements for an instance group,
+    # including configurations for both Spot and On-Demand capacity types.
+    #
+    # @!attribute [rw] spot
+    #   Configuration options specific to Spot instances.
+    #   @return [Types::ClusterSpotOptions]
+    #
+    # @!attribute [rw] on_demand
+    #   Configuration options specific to On-Demand instances.
+    #   @return [Types::ClusterOnDemandOptions]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterCapacityRequirements AWS API Documentation
+    #
+    class ClusterCapacityRequirements < Struct.new(
+      :spot,
+      :on_demand)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Defines the configuration for attaching an additional Amazon Elastic
     # Block Store (EBS) volume to each instance of the SageMaker HyperPod
     # cluster instance group. To learn more, see [SageMaker HyperPod release
@@ -5329,6 +5349,12 @@ module Aws::SageMaker
     #   of a SageMaker HyperPod cluster.
     #   @return [Integer]
     #
+    # @!attribute [rw] min_count
+    #   The minimum number of instances that must be available in the
+    #   instance group of a SageMaker HyperPod cluster before it transitions
+    #   to `InService` status.
+    #   @return [Integer]
+    #
     # @!attribute [rw] instance_group_name
     #   The name of the instance group of a SageMaker HyperPod cluster.
     #   @return [String]
@@ -5423,6 +5449,23 @@ module Aws::SageMaker
     #   group.
     #   @return [String]
     #
+    # @!attribute [rw] active_operations
+    #   A map indicating active operations currently in progress for the
+    #   instance group of a SageMaker HyperPod cluster. When there is a
+    #   scaling operation in progress, this map contains a key `Scaling`
+    #   with value 1.
+    #   @return [Hash<String,Integer>]
+    #
+    # @!attribute [rw] kubernetes_config
+    #   The Kubernetes configuration for the instance group that contains
+    #   labels and taints to be applied for the nodes in this instance
+    #   group.
+    #   @return [Types::ClusterKubernetesConfigDetails]
+    #
+    # @!attribute [rw] capacity_requirements
+    #   The instance capacity requirements for the instance group.
+    #   @return [Types::ClusterCapacityRequirements]
+    #
     # @!attribute [rw] target_state_count
     #   The number of nodes running a specific image ID since the last
     #   software update request.
@@ -5441,6 +5484,7 @@ module Aws::SageMaker
     class ClusterInstanceGroupDetails < Struct.new(
       :current_count,
       :target_count,
+      :min_count,
       :instance_group_name,
       :instance_type,
       :life_cycle_config,
@@ -5455,6 +5499,9 @@ module Aws::SageMaker
       :scheduled_update_config,
       :current_image_id,
       :desired_image_id,
+      :active_operations,
+      :kubernetes_config,
+      :capacity_requirements,
       :target_state_count,
       :software_update_status,
       :active_software_update_config)
@@ -5469,6 +5516,16 @@ module Aws::SageMaker
     #   SageMaker HyperPod cluster.
     #   @return [Integer]
     #
+    # @!attribute [rw] min_instance_count
+    #   Defines the minimum number of instances required for an instance
+    #   group to become `InService`. If this threshold isn't met within 3
+    #   hours, the instance group rolls back to its previous state - zero
+    #   instances for new instance groups, or previous settings for existing
+    #   instance groups. `MinInstanceCount` only affects the initial
+    #   transition to `InService` and does not guarantee maintaining this
+    #   minimum afterward.
+    #   @return [Integer]
+    #
     # @!attribute [rw] instance_group_name
     #   Specifies the name of the instance group.
     #   @return [String]
@@ -5596,10 +5653,22 @@ module Aws::SageMaker
     #   patched with the specified image.
     #   @return [String]
     #
+    # @!attribute [rw] kubernetes_config
+    #   Specifies the Kubernetes configuration for the instance group. You
+    #   describe what you want the labels and taints to look like, and the
+    #   cluster works to reconcile the actual state with the declared state
+    #   for nodes in this instance group.
+    #   @return [Types::ClusterKubernetesConfig]
+    #
+    # @!attribute [rw] capacity_requirements
+    #   Specifies the capacity requirements for the instance group.
+    #   @return [Types::ClusterCapacityRequirements]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterInstanceGroupSpecification AWS API Documentation
     #
     class ClusterInstanceGroupSpecification < Struct.new(
       :instance_count,
+      :min_instance_count,
       :instance_group_name,
       :instance_type,
       :life_cycle_config,
@@ -5610,7 +5679,9 @@ module Aws::SageMaker
       :training_plan_arn,
       :override_vpc_config,
       :scheduled_update_config,
-      :image_id)
+      :image_id,
+      :kubernetes_config,
+      :capacity_requirements)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -5690,6 +5761,113 @@ module Aws::SageMaker
       class Unknown < ClusterInstanceStorageConfig; end
     end
 
+    # Kubernetes configuration that specifies labels and taints to be
+    # applied to cluster nodes in an instance group.
+    #
+    # @!attribute [rw] labels
+    #   Key-value pairs of labels to be applied to cluster nodes.
+    #   @return [Hash<String,String>]
+    #
+    # @!attribute [rw] taints
+    #   List of taints to be applied to cluster nodes.
+    #   @return [Array<Types::ClusterKubernetesTaint>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterKubernetesConfig AWS API Documentation
+    #
+    class ClusterKubernetesConfig < Struct.new(
+      :labels,
+      :taints)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Detailed Kubernetes configuration showing both the current and desired
+    # state of labels and taints for cluster nodes.
+    #
+    # @!attribute [rw] current_labels
+    #   The current labels applied to cluster nodes of an instance group.
+    #   @return [Hash<String,String>]
+    #
+    # @!attribute [rw] desired_labels
+    #   The desired labels to be applied to cluster nodes of an instance
+    #   group.
+    #   @return [Hash<String,String>]
+    #
+    # @!attribute [rw] current_taints
+    #   The current taints applied to cluster nodes of an instance group.
+    #   @return [Array<Types::ClusterKubernetesTaint>]
+    #
+    # @!attribute [rw] desired_taints
+    #   The desired taints to be applied to cluster nodes of an instance
+    #   group.
+    #   @return [Array<Types::ClusterKubernetesTaint>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterKubernetesConfigDetails AWS API Documentation
+    #
+    class ClusterKubernetesConfigDetails < Struct.new(
+      :current_labels,
+      :desired_labels,
+      :current_taints,
+      :desired_taints)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Node-specific Kubernetes configuration showing both current and
+    # desired state of labels and taints for an individual cluster node.
+    #
+    # @!attribute [rw] current_labels
+    #   The current labels applied to the cluster node.
+    #   @return [Hash<String,String>]
+    #
+    # @!attribute [rw] desired_labels
+    #   The desired labels to be applied to the cluster node.
+    #   @return [Hash<String,String>]
+    #
+    # @!attribute [rw] current_taints
+    #   The current taints applied to the cluster node.
+    #   @return [Array<Types::ClusterKubernetesTaint>]
+    #
+    # @!attribute [rw] desired_taints
+    #   The desired taints to be applied to the cluster node.
+    #   @return [Array<Types::ClusterKubernetesTaint>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterKubernetesConfigNodeDetails AWS API Documentation
+    #
+    class ClusterKubernetesConfigNodeDetails < Struct.new(
+      :current_labels,
+      :desired_labels,
+      :current_taints,
+      :desired_taints)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A Kubernetes taint that can be applied to cluster nodes.
+    #
+    # @!attribute [rw] key
+    #   The key of the taint.
+    #   @return [String]
+    #
+    # @!attribute [rw] value
+    #   The value of the taint.
+    #   @return [String]
+    #
+    # @!attribute [rw] effect
+    #   The effect of the taint. Valid values are `NoSchedule`,
+    #   `PreferNoSchedule`, and `NoExecute`.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterKubernetesTaint AWS API Documentation
+    #
+    class ClusterKubernetesTaint < Struct.new(
+      :key,
+      :value,
+      :effect)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The lifecycle configuration for a SageMaker HyperPod cluster.
     #
     # @!attribute [rw] source_s3_uri
@@ -5839,6 +6017,19 @@ module Aws::SageMaker
     #   Contains information about the UltraServer.
     #   @return [Types::UltraServerInfo]
     #
+    # @!attribute [rw] kubernetes_config
+    #   The Kubernetes configuration applied to this node, showing both the
+    #   current and desired state of labels and taints. The cluster works to
+    #   reconcile the actual state with the declared state.
+    #   @return [Types::ClusterKubernetesConfigNodeDetails]
+    #
+    # @!attribute [rw] capacity_type
+    #   The capacity type of the node. Valid values are `OnDemand` and
+    #   `Spot`. When set to `OnDemand`, the node is launched as an On-Demand
+    #   instance. When set to `Spot`, the node is launched as a Spot
+    #   instance.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterNodeDetails AWS API Documentation
     #
     class ClusterNodeDetails < Struct.new(
@@ -5859,7 +6050,9 @@ module Aws::SageMaker
       :placement,
       :current_image_id,
       :desired_image_id,
-      :ultra_server_info)
+      :ultra_server_info,
+      :kubernetes_config,
+      :capacity_type)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -5925,6 +6118,14 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Configuration options specific to On-Demand instances.
+    #
+    # @api private
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterOnDemandOptions AWS API Documentation
+    #
+    class ClusterOnDemandOptions < Aws::EmptyStructure; end
+
     # The type of orchestrator used for the SageMaker HyperPod cluster.
     #
     # @!attribute [rw] eks
@@ -6221,6 +6422,14 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Configuration options specific to Spot instances.
+    #
+    # @api private
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterSpotOptions AWS API Documentation
+    #
+    class ClusterSpotOptions < Aws::EmptyStructure; end
+
     # Lists a summary of the properties of a SageMaker HyperPod cluster.
     #
     # @!attribute [rw] cluster_arn
@@ -11189,6 +11398,10 @@ module Aws::SageMaker
     #   with the optimization job.
     #   @return [String]
     #
+    # @!attribute [rw] max_instance_count
+    #   The maximum number of instances to use for the optimization job.
+    #   @return [Integer]
+    #
     # @!attribute [rw] optimization_environment
     #   The environment variables to set in the model container.
     #   @return [Hash<String,String>]
@@ -11248,6 +11461,7 @@ module Aws::SageMaker
       :role_arn,
       :model_source,
       :deployment_instance_type,
+      :max_instance_count,
       :optimization_environment,
       :optimization_configs,
       :output_config,
@@ -19760,6 +19974,10 @@ module Aws::SageMaker
     #   with the optimization job.
     #   @return [String]
     #
+    # @!attribute [rw] max_instance_count
+    #   The maximum number of instances to use for the optimization job.
+    #   @return [Integer]
+    #
     # @!attribute [rw] optimization_configs
     #   Settings for each of the optimization techniques that the job
     #   applies.
@@ -19820,6 +20038,7 @@ module Aws::SageMaker
       :model_source,
       :optimization_environment,
       :deployment_instance_type,
+      :max_instance_count,
       :optimization_configs,
       :output_config,
       :optimization_output,
@@ -29039,6 +29258,10 @@ module Aws::SageMaker
     #   The desired number of instances for the group after scaling.
     #   @return [Integer]
     #
+    # @!attribute [rw] min_count
+    #   Minimum instance count of the instance group.
+    #   @return [Integer]
+    #
     # @!attribute [rw] failure_message
     #   An error message describing why the scaling operation failed, if
     #   applicable.
@@ -29049,6 +29272,7 @@ module Aws::SageMaker
     class InstanceGroupScalingMetadata < Struct.new(
       :instance_count,
       :target_count,
+      :min_count,
       :failure_message)
       SENSITIVE = []
       include Aws::Structure
@@ -38430,6 +38654,53 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Settings for the model speculative decoding technique that's applied
+    # by a model optimization job.
+    #
+    # @!attribute [rw] technique
+    #   The speculative decoding technique to apply during model
+    #   optimization.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_data_source
+    #   The location of the training data to use for speculative decoding.
+    #   The data must be formatted as ShareGPT, OpenAI Completions or OpenAI
+    #   Chat Completions. The input can also be unencrypted captured data
+    #   from a SageMaker endpoint as long as the endpoint uses one of the
+    #   above formats.
+    #   @return [Types::ModelSpeculativeDecodingTrainingDataSource]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ModelSpeculativeDecodingConfig AWS API Documentation
+    #
+    class ModelSpeculativeDecodingConfig < Struct.new(
+      :technique,
+      :training_data_source)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Contains information about the training data source for speculative
+    # decoding.
+    #
+    # @!attribute [rw] s3_uri
+    #   The Amazon S3 URI that points to the training data for speculative
+    #   decoding.
+    #   @return [String]
+    #
+    # @!attribute [rw] s3_data_type
+    #   The type of data stored in the Amazon S3 location. Valid values are
+    #   `S3Prefix` or `ManifestFile`.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ModelSpeculativeDecodingTrainingDataSource AWS API Documentation
+    #
+    class ModelSpeculativeDecodingTrainingDataSource < Struct.new(
+      :s3_uri,
+      :s3_data_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Metadata for Model steps.
     #
     # @!attribute [rw] arn
@@ -40042,12 +40313,18 @@ module Aws::SageMaker
     #   optimization job.
     #   @return [Types::ModelShardingConfig]
     #
+    # @!attribute [rw] model_speculative_decoding_config
+    #   Settings for the model speculative decoding technique that's
+    #   applied by a model optimization job.
+    #   @return [Types::ModelSpeculativeDecodingConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/OptimizationConfig AWS API Documentation
     #
     class OptimizationConfig < Struct.new(
       :model_quantization_config,
       :model_compilation_config,
       :model_sharding_config,
+      :model_speculative_decoding_config,
       :unknown)
       SENSITIVE = []
       include Aws::Structure
@@ -40056,6 +40333,7 @@ module Aws::SageMaker
       class ModelQuantizationConfig < OptimizationConfig; end
       class ModelCompilationConfig < OptimizationConfig; end
       class ModelShardingConfig < OptimizationConfig; end
+      class ModelSpeculativeDecodingConfig < OptimizationConfig; end
       class Unknown < OptimizationConfig; end
     end
 
@@ -40066,10 +40344,16 @@ module Aws::SageMaker
     #   optimization job.
     #   @return [Types::OptimizationJobModelSourceS3]
     #
+    # @!attribute [rw] sage_maker_model
+    #   The name of an existing SageMaker model to optimize with an
+    #   optimization job.
+    #   @return [Types::OptimizationSageMakerModel]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/OptimizationJobModelSource AWS API Documentation
     #
     class OptimizationJobModelSource < Struct.new(
-      :s3)
+      :s3,
+      :sage_maker_model)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -40111,11 +40395,17 @@ module Aws::SageMaker
     #   create with an optimization job.
     #   @return [String]
     #
+    # @!attribute [rw] sage_maker_model
+    #   The name of a SageMaker model to use as the output destination for
+    #   an optimization job.
+    #   @return [Types::OptimizationSageMakerModel]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/OptimizationJobOutputConfig AWS API Documentation
     #
     class OptimizationJobOutputConfig < Struct.new(
       :kms_key_id,
-      :s3_output_location)
+      :s3_output_location,
+      :sage_maker_model)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -40156,6 +40446,10 @@ module Aws::SageMaker
     #   with the optimization job.
     #   @return [String]
     #
+    # @!attribute [rw] max_instance_count
+    #   The maximum number of instances to use for the optimization job.
+    #   @return [Integer]
+    #
     # @!attribute [rw] optimization_types
     #   The optimization techniques that are applied by the optimization
     #   job.
@@ -40172,6 +40466,7 @@ module Aws::SageMaker
       :optimization_end_time,
       :last_modified_time,
       :deployment_instance_type,
+      :max_instance_count,
       :optimization_types)
       SENSITIVE = []
       include Aws::Structure
@@ -40213,6 +40508,21 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # A SageMaker model to use as the source or destination for an
+    # optimization job.
+    #
+    # @!attribute [rw] model_name
+    #   The name of a SageMaker model.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/OptimizationSageMakerModel AWS API Documentation
+    #
+    class OptimizationSageMakerModel < Struct.new(
+      :model_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # A VPC in Amazon VPC that's accessible to an optimized that you create
     # with an optimization job. You can control access to and from your
     # resources by configuring a VPC. For more information, see [Give

data/lib/aws-sdk-sagemaker.rb

@@ -55,7 +55,7 @@ module Aws::SageMaker
   autoload :EndpointProvider, 'aws-sdk-sagemaker/endpoint_provider'
   autoload :Endpoints, 'aws-sdk-sagemaker/endpoints'
 
-  GEM_VERSION = '1.340.0'
+  GEM_VERSION = '1.341.0'
 
 end
 

data/sig/client.rbs

@@ -806,6 +806,7 @@ module Aws
                             ?instance_groups: Array[
                               {
                                 instance_count: ::Integer,
+                                min_instance_count: ::Integer?,
                                 instance_group_name: ::String,
                                 instance_type: ("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.p6e-gb200.36xlarge" | "ml.trn1.32xlarge" | "ml.trn1n.32xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.c5.large" | "ml.c5.xlarge" | "ml.c5.2xlarge" | "ml.c5.4xlarge" | "ml.c5.9xlarge" | "ml.c5.12xlarge" | "ml.c5.18xlarge" | "ml.c5.24xlarge" | "ml.c5n.large" | "ml.c5n.2xlarge" | "ml.c5n.4xlarge" | "ml.c5n.9xlarge" | "ml.c5n.18xlarge" | "ml.m5.large" | "ml.m5.xlarge" | "ml.m5.2xlarge" | "ml.m5.4xlarge" | "ml.m5.8xlarge" | "ml.m5.12xlarge" | "ml.m5.16xlarge" | "ml.m5.24xlarge" | "ml.t3.medium" | "ml.t3.large" | "ml.t3.xlarge" | "ml.t3.2xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.16xlarge" | "ml.g6.12xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge" | "ml.gr6.4xlarge" | "ml.gr6.8xlarge" | "ml.g6e.xlarge" | "ml.g6e.2xlarge" | "ml.g6e.4xlarge" | "ml.g6e.8xlarge" | "ml.g6e.16xlarge" | "ml.g6e.12xlarge" | "ml.g6e.24xlarge" | "ml.g6e.48xlarge" | "ml.p5e.48xlarge" | "ml.p5en.48xlarge" | "ml.p6-b200.48xlarge" | "ml.trn2.3xlarge" | "ml.trn2.48xlarge" | "ml.c6i.large" | "ml.c6i.xlarge" | "ml.c6i.2xlarge" | "ml.c6i.4xlarge" | "ml.c6i.8xlarge" | "ml.c6i.12xlarge" | "ml.c6i.16xlarge" | "ml.c6i.24xlarge" | "ml.c6i.32xlarge" | "ml.m6i.large" | "ml.m6i.xlarge" | "ml.m6i.2xlarge" | "ml.m6i.4xlarge" | "ml.m6i.8xlarge" | "ml.m6i.12xlarge" | "ml.m6i.16xlarge" | "ml.m6i.24xlarge" | "ml.m6i.32xlarge" | "ml.r6i.large" | "ml.r6i.xlarge" | "ml.r6i.2xlarge" | "ml.r6i.4xlarge" | "ml.r6i.8xlarge" | "ml.r6i.12xlarge" | "ml.r6i.16xlarge" | "ml.r6i.24xlarge" | "ml.r6i.32xlarge" | "ml.i3en.large" | "ml.i3en.xlarge" | "ml.i3en.2xlarge" | "ml.i3en.3xlarge" | "ml.i3en.6xlarge" | "ml.i3en.12xlarge" | "ml.i3en.24xlarge" | "ml.m7i.large" | "ml.m7i.xlarge" | "ml.m7i.2xlarge" | "ml.m7i.4xlarge" | "ml.m7i.8xlarge" | "ml.m7i.12xlarge" | "ml.m7i.16xlarge" | "ml.m7i.24xlarge" | "ml.m7i.48xlarge" | "ml.r7i.large" | "ml.r7i.xlarge" | "ml.r7i.2xlarge" | "ml.r7i.4xlarge" | "ml.r7i.8xlarge" | "ml.r7i.12xlarge" | "ml.r7i.16xlarge" | "ml.r7i.24xlarge" | "ml.r7i.48xlarge"),
                                 life_cycle_config: {
@@ -850,7 +851,23 @@ module Aws
                                     ]?
                                   }?
                                 }?,
-                                image_id: ::String?
+                                image_id: ::String?,
+                                kubernetes_config: {
+                                  labels: Hash[::String, ::String]?,
+                                  taints: Array[
+                                    {
+                                      key: ::String,
+                                      value: ::String?,
+                                      effect: ("NoSchedule" | "PreferNoSchedule" | "NoExecute")
+                                    },
+                                  ]?
+                                }?,
+                                capacity_requirements: {
+                                  spot: {
+                                  }?,
+                                  on_demand: {
+                                  }?
+                                }?
                               },
                             ],
                             ?restricted_instance_groups: Array[
@@ -3788,9 +3805,13 @@ module Aws
                                          model_access_config: {
                                            accept_eula: bool
                                          }?
+                                       }?,
+                                       sage_maker_model: {
+                                         model_name: ::String?
                                        }?
                                      },
-                                     deployment_instance_type: ("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge" | "ml.g6e.xlarge" | "ml.g6e.2xlarge" | "ml.g6e.4xlarge" | "ml.g6e.8xlarge" | "ml.g6e.12xlarge" | "ml.g6e.16xlarge" | "ml.g6e.24xlarge" | "ml.g6e.48xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.trn1n.32xlarge"),
+                                     deployment_instance_type: ("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.p5e.48xlarge" | "ml.p5en.48xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge" | "ml.g6e.xlarge" | "ml.g6e.2xlarge" | "ml.g6e.4xlarge" | "ml.g6e.8xlarge" | "ml.g6e.12xlarge" | "ml.g6e.16xlarge" | "ml.g6e.24xlarge" | "ml.g6e.48xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.trn1n.32xlarge"),
+                                     ?max_instance_count: ::Integer,
                                      ?optimization_environment: Hash[::String, ::String],
                                      optimization_configs: Array[
                                        {
@@ -3805,12 +3826,22 @@ module Aws
                                          model_sharding_config: {
                                            image: ::String?,
                                            override_environment: Hash[::String, ::String]?
+                                         }?,
+                                         model_speculative_decoding_config: {
+                                           technique: ("EAGLE"),
+                                           training_data_source: {
+                                             s3_uri: ::String,
+                                             s3_data_type: ("S3Prefix" | "ManifestFile")
+                                           }?
                                          }?
                                        },
                                      ],
                                      output_config: {
                                        kms_key_id: ::String?,
-                                       s3_output_location: ::String
+                                       s3_output_location: ::String,
+                                       sage_maker_model: {
+                                         model_name: ::String?
+                                       }?
                                      },
                                      stopping_condition: {
                                        max_runtime_in_seconds: ::Integer?,
@@ -6422,7 +6453,8 @@ module Aws
         def optimization_job_name: () -> ::String
         def model_source: () -> Types::OptimizationJobModelSource
         def optimization_environment: () -> ::Hash[::String, ::String]
-        def deployment_instance_type: () -> ("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge" | "ml.g6e.xlarge" | "ml.g6e.2xlarge" | "ml.g6e.4xlarge" | "ml.g6e.8xlarge" | "ml.g6e.12xlarge" | "ml.g6e.16xlarge" | "ml.g6e.24xlarge" | "ml.g6e.48xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.trn1n.32xlarge")
+        def deployment_instance_type: () -> ("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.p5e.48xlarge" | "ml.p5en.48xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.12xlarge" | "ml.g6.16xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge" | "ml.g6e.xlarge" | "ml.g6e.2xlarge" | "ml.g6e.4xlarge" | "ml.g6e.8xlarge" | "ml.g6e.12xlarge" | "ml.g6e.16xlarge" | "ml.g6e.24xlarge" | "ml.g6e.48xlarge" | "ml.inf2.xlarge" | "ml.inf2.8xlarge" | "ml.inf2.24xlarge" | "ml.inf2.48xlarge" | "ml.trn1.2xlarge" | "ml.trn1.32xlarge" | "ml.trn1n.32xlarge")
+        def max_instance_count: () -> ::Integer
         def optimization_configs: () -> ::Array[Types::OptimizationConfig]
         def output_config: () -> Types::OptimizationJobOutputConfig
         def optimization_output: () -> Types::OptimizationOutput
@@ -8986,6 +9018,7 @@ module Aws
                             ?instance_groups: Array[
                               {
                                 instance_count: ::Integer,
+                                min_instance_count: ::Integer?,
                                 instance_group_name: ::String,
                                 instance_type: ("ml.p4d.24xlarge" | "ml.p4de.24xlarge" | "ml.p5.48xlarge" | "ml.p6e-gb200.36xlarge" | "ml.trn1.32xlarge" | "ml.trn1n.32xlarge" | "ml.g5.xlarge" | "ml.g5.2xlarge" | "ml.g5.4xlarge" | "ml.g5.8xlarge" | "ml.g5.12xlarge" | "ml.g5.16xlarge" | "ml.g5.24xlarge" | "ml.g5.48xlarge" | "ml.c5.large" | "ml.c5.xlarge" | "ml.c5.2xlarge" | "ml.c5.4xlarge" | "ml.c5.9xlarge" | "ml.c5.12xlarge" | "ml.c5.18xlarge" | "ml.c5.24xlarge" | "ml.c5n.large" | "ml.c5n.2xlarge" | "ml.c5n.4xlarge" | "ml.c5n.9xlarge" | "ml.c5n.18xlarge" | "ml.m5.large" | "ml.m5.xlarge" | "ml.m5.2xlarge" | "ml.m5.4xlarge" | "ml.m5.8xlarge" | "ml.m5.12xlarge" | "ml.m5.16xlarge" | "ml.m5.24xlarge" | "ml.t3.medium" | "ml.t3.large" | "ml.t3.xlarge" | "ml.t3.2xlarge" | "ml.g6.xlarge" | "ml.g6.2xlarge" | "ml.g6.4xlarge" | "ml.g6.8xlarge" | "ml.g6.16xlarge" | "ml.g6.12xlarge" | "ml.g6.24xlarge" | "ml.g6.48xlarge" | "ml.gr6.4xlarge" | "ml.gr6.8xlarge" | "ml.g6e.xlarge" | "ml.g6e.2xlarge" | "ml.g6e.4xlarge" | "ml.g6e.8xlarge" | "ml.g6e.16xlarge" | "ml.g6e.12xlarge" | "ml.g6e.24xlarge" | "ml.g6e.48xlarge" | "ml.p5e.48xlarge" | "ml.p5en.48xlarge" | "ml.p6-b200.48xlarge" | "ml.trn2.3xlarge" | "ml.trn2.48xlarge" | "ml.c6i.large" | "ml.c6i.xlarge" | "ml.c6i.2xlarge" | "ml.c6i.4xlarge" | "ml.c6i.8xlarge" | "ml.c6i.12xlarge" | "ml.c6i.16xlarge" | "ml.c6i.24xlarge" | "ml.c6i.32xlarge" | "ml.m6i.large" | "ml.m6i.xlarge" | "ml.m6i.2xlarge" | "ml.m6i.4xlarge" | "ml.m6i.8xlarge" | "ml.m6i.12xlarge" | "ml.m6i.16xlarge" | "ml.m6i.24xlarge" | "ml.m6i.32xlarge" | "ml.r6i.large" | "ml.r6i.xlarge" | "ml.r6i.2xlarge" | "ml.r6i.4xlarge" | "ml.r6i.8xlarge" | "ml.r6i.12xlarge" | "ml.r6i.16xlarge" | "ml.r6i.24xlarge" | "ml.r6i.32xlarge" | "ml.i3en.large" | "ml.i3en.xlarge" | "ml.i3en.2xlarge" | "ml.i3en.3xlarge" | "ml.i3en.6xlarge" | "ml.i3en.12xlarge" | "ml.i3en.24xlarge" | "ml.m7i.large" | "ml.m7i.xlarge" | "ml.m7i.2xlarge" | "ml.m7i.4xlarge" | "ml.m7i.8xlarge" | "ml.m7i.12xlarge" | "ml.m7i.16xlarge" | "ml.m7i.24xlarge" | "ml.m7i.48xlarge" | "ml.r7i.large" | "ml.r7i.xlarge" | "ml.r7i.2xlarge" | "ml.r7i.4xlarge" | "ml.r7i.8xlarge" | "ml.r7i.12xlarge" | "ml.r7i.16xlarge" | "ml.r7i.24xlarge" | "ml.r7i.48xlarge"),
                                 life_cycle_config: {
@@ -9030,7 +9063,23 @@ module Aws
                                     ]?
                                   }?
                                 }?,
-                                image_id: ::String?
+                                image_id: ::String?,
+                                kubernetes_config: {
+                                  labels: Hash[::String, ::String]?,
+                                  taints: Array[
+                                    {
+                                      key: ::String,
+                                      value: ::String?,
+                                      effect: ("NoSchedule" | "PreferNoSchedule" | "NoExecute")
+                                    },
+                                  ]?
+                                }?,
+                                capacity_requirements: {
+                                  spot: {
+                                  }?,
+                                  on_demand: {
+                                  }?
+                                }?
                               },
                             ],
                             ?restricted_instance_groups: Array[