aws-sdk-sagemaker 1.276.0 → 1.278.0

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

@@ -310,6 +310,13 @@ module Aws::SageMaker
     # Specifies the training algorithm to use in a [CreateTrainingJob][1]
     # request.
     #
+    # SageMaker uses its own SageMaker account credentials to pull and
+    # access built-in algorithms so built-in algorithms are universally
+    # accessible across all Amazon Web Services accounts. As a result,
+    # built-in algorithms have standard, unrestricted access. You cannot
+    # restrict built-in algorithms using IAM roles. Use custom algorithms if
+    # you require specific access controls.
+    #
     # For more information about algorithms provided by SageMaker, see
     # [Algorithms][2]. For information about using your own algorithms, see
     # [Using Your Own Algorithms with Amazon SageMaker][3].
@@ -1297,7 +1304,7 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # Details about an Amazon SageMaker app.
+    # Details about an Amazon SageMaker AI app.
     #
     # @!attribute [rw] domain_id
     #   The domain ID.
@@ -1328,7 +1335,7 @@ module Aws::SageMaker
     #   @return [Time]
     #
     # @!attribute [rw] resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
@@ -1347,7 +1354,7 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # The configuration for running a SageMaker image as a KernelGateway
+    # The configuration for running a SageMaker AI image as a KernelGateway
     # app.
     #
     # @!attribute [rw] app_image_config_arn
@@ -1368,7 +1375,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] kernel_gateway_image_config
     #   The configuration for the file system and kernels in the SageMaker
-    #   image.
+    #   AI image.
     #   @return [Types::KernelGatewayImageConfig]
     #
     # @!attribute [rw] jupyter_lab_app_image_config
@@ -2720,7 +2727,7 @@ module Aws::SageMaker
     #   The data type.
     #
     #   * If you choose `S3Prefix`, `S3Uri` identifies a key name prefix.
-    #     SageMaker uses all objects that match the specified key name
+    #     SageMaker AI uses all objects that match the specified key name
     #     prefix for model training.
     #
     #     The `S3Prefix` should have the following format:
@@ -2729,7 +2736,7 @@ module Aws::SageMaker
     #
     #   * If you choose `ManifestFile`, `S3Uri` identifies an object that is
     #     a manifest file containing a list of object keys that you want
-    #     SageMaker to use for model training.
+    #     SageMaker AI to use for model training.
     #
     #     A `ManifestFile` should have the format shown below:
     #
@@ -3506,16 +3513,16 @@ module Aws::SageMaker
     end
 
     # Configuration specifying how to treat different headers. If no headers
-    # are specified Amazon SageMaker will by default base64 encode when
+    # are specified Amazon SageMaker AI will by default base64 encode when
     # capturing the data.
     #
     # @!attribute [rw] csv_content_types
-    #   The list of all content type headers that Amazon SageMaker will
+    #   The list of all content type headers that Amazon SageMaker AI will
     #   treat as CSV and capture accordingly.
     #   @return [Array<String>]
     #
     # @!attribute [rw] json_content_types
-    #   The list of all content type headers that SageMaker will treat as
+    #   The list of all content type headers that SageMaker AI will treat as
     #   JSON and capture accordingly.
     #   @return [Array<String>]
     #
@@ -4227,6 +4234,38 @@ module Aws::SageMaker
     #   when the cluster instance group is created or updated.
     #   @return [Array<String>]
     #
+    # @!attribute [rw] status
+    #   The current status of the cluster instance group.
+    #
+    #   * `InService`: The instance group is active and healthy.
+    #
+    #   * `Creating`: The instance group is being provisioned.
+    #
+    #   * `Updating`: The instance group is being updated.
+    #
+    #   * `Failed`: The instance group has failed to provision or is no
+    #     longer healthy.
+    #
+    #   * `Degraded`: The instance group is degraded, meaning that some
+    #     instances have failed to provision or are no longer healthy.
+    #
+    #   * `Deleting`: The instance group is being deleted.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan associated with
+    #   this cluster instance group.
+    #
+    #   For more information about how to reserve GPU capacity for your
+    #   SageMaker HyperPod clusters using Amazon SageMaker Training Plan,
+    #   see ` CreateTrainingPlan `.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_plan_status
+    #   The current status of the training plan associated with this cluster
+    #   instance group.
+    #   @return [String]
+    #
     # @!attribute [rw] override_vpc_config
     #   Specifies an Amazon Virtual Private Cloud (VPC) that your SageMaker
     #   jobs, hosted models, and compute resources have access to. You can
@@ -4251,6 +4290,9 @@ module Aws::SageMaker
       :threads_per_core,
       :instance_storage_configs,
       :on_start_deep_health_checks,
+      :status,
+      :training_plan_arn,
+      :training_plan_status,
       :override_vpc_config)
       SENSITIVE = []
       include Aws::Structure
@@ -4303,6 +4345,15 @@ module Aws::SageMaker
     #   when the cluster instance group is created or updated.
     #   @return [Array<String>]
     #
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan to use for this
+    #   cluster instance group.
+    #
+    #   For more information about how to reserve GPU capacity for your
+    #   SageMaker HyperPod clusters using Amazon SageMaker Training Plan,
+    #   see ` CreateTrainingPlan `.
+    #   @return [String]
+    #
     # @!attribute [rw] override_vpc_config
     #   Specifies an Amazon Virtual Private Cloud (VPC) that your SageMaker
     #   jobs, hosted models, and compute resources have access to. You can
@@ -4326,6 +4377,7 @@ module Aws::SageMaker
       :threads_per_core,
       :instance_storage_configs,
       :on_start_deep_health_checks,
+      :training_plan_arn,
       :override_vpc_config)
       SENSITIVE = []
       include Aws::Structure
@@ -4583,6 +4635,55 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Summary of the cluster policy.
+    #
+    # @!attribute [rw] cluster_scheduler_config_arn
+    #   ARN of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_scheduler_config_id
+    #   ID of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_scheduler_config_version
+    #   Version of the cluster policy.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] name
+    #   Name of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   Creation time of the cluster policy.
+    #   @return [Time]
+    #
+    # @!attribute [rw] last_modified_time
+    #   Last modified time of the cluster policy.
+    #   @return [Time]
+    #
+    # @!attribute [rw] status
+    #   Status of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   ARN of the cluster.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterSchedulerConfigSummary AWS API Documentation
+    #
+    class ClusterSchedulerConfigSummary < Struct.new(
+      :cluster_scheduler_config_arn,
+      :cluster_scheduler_config_id,
+      :cluster_scheduler_config_version,
+      :name,
+      :creation_time,
+      :last_modified_time,
+      :status,
+      :cluster_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Lists a summary of the properties of a SageMaker HyperPod cluster.
     #
     # @!attribute [rw] cluster_arn
@@ -4601,13 +4702,23 @@ module Aws::SageMaker
     #   The status of the SageMaker HyperPod cluster.
     #   @return [String]
     #
+    # @!attribute [rw] training_plan_arns
+    #   A list of Amazon Resource Names (ARNs) of the training plans
+    #   associated with this cluster.
+    #
+    #   For more information about how to reserve GPU capacity for your
+    #   SageMaker HyperPod clusters using Amazon SageMaker Training Plan,
+    #   see ` CreateTrainingPlan `.
+    #   @return [Array<String>]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ClusterSummary AWS API Documentation
     #
     class ClusterSummary < Struct.new(
       :cluster_arn,
       :cluster_name,
       :creation_time,
-      :cluster_status)
+      :cluster_status,
+      :training_plan_arns)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -4618,7 +4729,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] file_system_config
     #   The Amazon Elastic File System storage configuration for a SageMaker
-    #   image.
+    #   AI image.
     #   @return [Types::FileSystemConfig]
     #
     # @!attribute [rw] container_config
@@ -4644,7 +4755,7 @@ module Aws::SageMaker
     # [1]: https://docs.aws.amazon.com/sagemaker/latest/dg/code-editor.html
     #
     # @!attribute [rw] default_resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
@@ -4681,7 +4792,7 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # A Git repository that SageMaker automatically displays to users for
+    # A Git repository that SageMaker AI automatically displays to users for
     # cloning in the JupyterServer application.
     #
     # @!attribute [rw] repository_url
@@ -4911,6 +5022,151 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Configuration of the compute allocation definition for an entity. This
+    # includes the resource sharing option and the setting to preempt low
+    # priority tasks.
+    #
+    # @!attribute [rw] compute_quota_resources
+    #   Allocate compute resources by instance types.
+    #   @return [Array<Types::ComputeQuotaResourceConfig>]
+    #
+    # @!attribute [rw] resource_sharing_config
+    #   Resource sharing configuration. This defines how an entity can lend
+    #   and borrow idle compute with other entities within the cluster.
+    #   @return [Types::ResourceSharingConfig]
+    #
+    # @!attribute [rw] preempt_team_tasks
+    #   Allows workloads from within an entity to preempt same-team
+    #   workloads. When set to `LowerPriority`, the entity's lower priority
+    #   tasks are preempted by their own higher priority tasks.
+    #
+    #   Default is `LowerPriority`.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ComputeQuotaConfig AWS API Documentation
+    #
+    class ComputeQuotaConfig < Struct.new(
+      :compute_quota_resources,
+      :resource_sharing_config,
+      :preempt_team_tasks)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Configuration of the resources used for the compute allocation
+    # definition.
+    #
+    # @!attribute [rw] instance_type
+    #   The instance type of the instance group for the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] count
+    #   The number of instances to add to the instance group of a SageMaker
+    #   HyperPod cluster.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ComputeQuotaResourceConfig AWS API Documentation
+    #
+    class ComputeQuotaResourceConfig < Struct.new(
+      :instance_type,
+      :count)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Summary of the compute allocation definition.
+    #
+    # @!attribute [rw] compute_quota_arn
+    #   ARN of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_id
+    #   ID of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] name
+    #   Name of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_version
+    #   Version of the compute allocation definition.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] status
+    #   Status of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   ARN of the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_config
+    #   Configuration of the compute allocation definition. This includes
+    #   the resource sharing option, and the setting to preempt low priority
+    #   tasks.
+    #   @return [Types::ComputeQuotaConfig]
+    #
+    # @!attribute [rw] compute_quota_target
+    #   The target entity to allocate compute resources to.
+    #   @return [Types::ComputeQuotaTarget]
+    #
+    # @!attribute [rw] activation_state
+    #   The state of the compute allocation being described. Use to enable
+    #   or disable compute allocation.
+    #
+    #   Default is `Enabled`.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   Creation time of the compute allocation definition.
+    #   @return [Time]
+    #
+    # @!attribute [rw] last_modified_time
+    #   Last modified time of the compute allocation definition.
+    #   @return [Time]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ComputeQuotaSummary AWS API Documentation
+    #
+    class ComputeQuotaSummary < Struct.new(
+      :compute_quota_arn,
+      :compute_quota_id,
+      :name,
+      :compute_quota_version,
+      :status,
+      :cluster_arn,
+      :compute_quota_config,
+      :compute_quota_target,
+      :activation_state,
+      :creation_time,
+      :last_modified_time)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The target entity to allocate compute resources to.
+    #
+    # @!attribute [rw] team_name
+    #   Name of the team to allocate compute resources to.
+    #   @return [String]
+    #
+    # @!attribute [rw] fair_share_weight
+    #   Assigned entity fair-share weight. Idle compute will be shared
+    #   across entities based on these assigned weights. This weight is only
+    #   used when `FairShare` is enabled.
+    #
+    #   A weight of 0 is the lowest priority and 100 is the highest. Weight
+    #   0 is the default.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ComputeQuotaTarget AWS API Documentation
+    #
+    class ComputeQuotaTarget < Struct.new(
+      :team_name,
+      :fair_share_weight)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Metadata for a Condition step.
     #
     # @!attribute [rw] outcome
@@ -5524,7 +5780,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] resource_spec
     #   The instance type and the Amazon Resource Name (ARN) of the
-    #   SageMaker image created on the instance.
+    #   SageMaker AI image created on the instance.
     #
     #   <note markdown="1"> The value of `InstanceType` passed as part of the `ResourceSpec` in
     #   the `CreateApp` call overrides the value passed as part of the
@@ -5931,6 +6187,55 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] name
+    #   Name for the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   ARN of the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] scheduler_config
+    #   Configuration about the monitoring schedule.
+    #   @return [Types::SchedulerConfig]
+    #
+    # @!attribute [rw] description
+    #   Description of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] tags
+    #   Tags of the cluster policy.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateClusterSchedulerConfigRequest AWS API Documentation
+    #
+    class CreateClusterSchedulerConfigRequest < Struct.new(
+      :name,
+      :cluster_arn,
+      :scheduler_config,
+      :description,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] cluster_scheduler_config_arn
+    #   ARN of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_scheduler_config_id
+    #   ID of the cluster policy.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateClusterSchedulerConfigResponse AWS API Documentation
+    #
+    class CreateClusterSchedulerConfigResponse < Struct.new(
+      :cluster_scheduler_config_arn,
+      :cluster_scheduler_config_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] code_repository_name
     #   The name of the Git repository. The name must have 1 to 63
     #   characters. Valid characters are a-z, A-Z, 0-9, and - (hyphen).
@@ -5983,9 +6288,10 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that enables Amazon
-    #   SageMaker to perform tasks on your behalf.
+    #   SageMaker AI to perform tasks on your behalf.
     #
-    #   During model compilation, Amazon SageMaker needs your permission to:
+    #   During model compilation, Amazon SageMaker AI needs your permission
+    #   to:
     #
     #   * Read input data from an S3 bucket
     #
@@ -5996,9 +6302,9 @@ module Aws::SageMaker
     #   * Publish metrics to Amazon CloudWatch
     #
     #   You grant permissions for all of these tasks to an IAM role. To pass
-    #   this role to Amazon SageMaker, the caller of this API must have the
-    #   `iam:PassRole` permission. For more information, see [Amazon
-    #   SageMaker Roles.][1]
+    #   this role to Amazon SageMaker AI, the caller of this API must have
+    #   the `iam:PassRole` permission. For more information, see [Amazon
+    #   SageMaker AI Roles.][1]
     #
     #
     #
@@ -6037,7 +6343,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] stopping_condition
     #   Specifies a limit to how long a model compilation job can run. When
-    #   the job reaches the time limit, Amazon SageMaker ends the
+    #   the job reaches the time limit, Amazon SageMaker AI ends the
     #   compilation job. Use this API to cap model training costs.
     #   @return [Types::StoppingCondition]
     #
@@ -6069,7 +6375,7 @@ module Aws::SageMaker
 
     # @!attribute [rw] compilation_job_arn
     #   If the action is successful, the service sends back an HTTP 200
-    #   response. Amazon SageMaker returns the following data in JSON
+    #   response. Amazon SageMaker AI returns the following data in JSON
     #   format:
     #
     #   * `CompilationJobArn`: The Amazon Resource Name (ARN) of the
@@ -6086,6 +6392,70 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] name
+    #   Name to the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] description
+    #   Description of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   ARN of the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_config
+    #   Configuration of the compute allocation definition. This includes
+    #   the resource sharing option, and the setting to preempt low priority
+    #   tasks.
+    #   @return [Types::ComputeQuotaConfig]
+    #
+    # @!attribute [rw] compute_quota_target
+    #   The target entity to allocate compute resources to.
+    #   @return [Types::ComputeQuotaTarget]
+    #
+    # @!attribute [rw] activation_state
+    #   The state of the compute allocation being described. Use to enable
+    #   or disable compute allocation.
+    #
+    #   Default is `Enabled`.
+    #   @return [String]
+    #
+    # @!attribute [rw] tags
+    #   Tags of the compute allocation definition.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateComputeQuotaRequest AWS API Documentation
+    #
+    class CreateComputeQuotaRequest < Struct.new(
+      :name,
+      :description,
+      :cluster_arn,
+      :compute_quota_config,
+      :compute_quota_target,
+      :activation_state,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] compute_quota_arn
+    #   ARN of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_id
+    #   ID of the compute allocation definition.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateComputeQuotaResponse AWS API Documentation
+    #
+    class CreateComputeQuotaResponse < Struct.new(
+      :compute_quota_arn,
+      :compute_quota_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] context_name
     #   The name of the context. Must be unique to your account in an Amazon
     #   Web Services Region.
@@ -6167,7 +6537,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform tasks on your behalf.
+    #   AI can assume to perform tasks on your behalf.
     #   @return [String]
     #
     # @!attribute [rw] stopping_condition
@@ -6304,7 +6674,7 @@ module Aws::SageMaker
     #   `PublicInternetOnly`.
     #
     #   * `PublicInternetOnly` - Non-EFS traffic is through a VPC managed by
-    #     Amazon SageMaker, which allows direct internet access
+    #     Amazon SageMaker AI, which allows direct internet access
     #
     #   * `VpcOnly` - All traffic is through the specified VPC and subnets
     #   @return [String]
@@ -6314,7 +6684,7 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] kms_key_id
-    #   SageMaker uses Amazon Web Services KMS to encrypt EFS and EBS
+    #   SageMaker AI uses Amazon Web Services KMS to encrypt EFS and EBS
     #   volumes attached to the domain with an Amazon Web Services managed
     #   key by default. For more control, specify a customer managed key.
     #   @return [String]
@@ -6504,7 +6874,7 @@ module Aws::SageMaker
     #   @return [Array<Types::ProductionVariant>]
     #
     # @!attribute [rw] data_capture_config
-    #   Configuration to control how SageMaker captures inference data.
+    #   Configuration to control how SageMaker AI captures inference data.
     #   @return [Types::DataCaptureConfig]
     #
     # @!attribute [rw] tags
@@ -6590,11 +6960,11 @@ module Aws::SageMaker
     #
     # @!attribute [rw] execution_role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform actions on your behalf. For more information,
-    #   see [SageMaker Roles][1].
+    #   AI can assume to perform actions on your behalf. For more
+    #   information, see [SageMaker AI Roles][1].
     #
-    #   <note markdown="1"> To be able to pass this role to Amazon SageMaker, the caller of this
-    #   action must have the `iam:PassRole` permission.
+    #   <note markdown="1"> To be able to pass this role to Amazon SageMaker AI, the caller of
+    #   this action must have the `iam:PassRole` permission.
     #
     #    </note>
     #
@@ -7280,7 +7650,7 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] role_arn
-    #   The ARN of an IAM role that enables Amazon SageMaker to perform
+    #   The ARN of an IAM role that enables Amazon SageMaker AI to perform
     #   tasks on your behalf.
     #   @return [String]
     #
@@ -7355,16 +7725,16 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] job_type
-    #   Indicates SageMaker job type compatibility.
+    #   Indicates SageMaker AI job type compatibility.
     #
-    #   * `TRAINING`: The image version is compatible with SageMaker
+    #   * `TRAINING`: The image version is compatible with SageMaker AI
     #     training jobs.
     #
-    #   * `INFERENCE`: The image version is compatible with SageMaker
+    #   * `INFERENCE`: The image version is compatible with SageMaker AI
     #     inference jobs.
     #
     #   * `NOTEBOOK_KERNEL`: The image version is compatible with SageMaker
-    #     notebook kernels.
+    #     AI notebook kernels.
     #   @return [String]
     #
     # @!attribute [rw] ml_framework
@@ -8046,7 +8416,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform tasks on your behalf.
+    #   AI can assume to perform tasks on your behalf.
     #   @return [String]
     #
     # @!attribute [rw] stopping_condition
@@ -8230,7 +8600,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform tasks on your behalf.
+    #   AI can assume to perform tasks on your behalf.
     #   @return [String]
     #
     # @!attribute [rw] stopping_condition
@@ -8666,7 +9036,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform tasks on your behalf.
+    #   AI can assume to perform tasks on your behalf.
     #   @return [String]
     #
     # @!attribute [rw] stopping_condition
@@ -8772,13 +9142,14 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   When you send any requests to Amazon Web Services resources from the
-    #   notebook instance, SageMaker assumes this role to perform tasks on
-    #   your behalf. You must grant this role necessary permissions so
-    #   SageMaker can perform these tasks. The policy must allow the
-    #   SageMaker service principal (sagemaker.amazonaws.com) permissions to
-    #   assume this role. For more information, see [SageMaker Roles][1].
-    #
-    #   <note markdown="1"> To be able to pass this role to SageMaker, the caller of this API
+    #   notebook instance, SageMaker AI assumes this role to perform tasks
+    #   on your behalf. You must grant this role necessary permissions so
+    #   SageMaker AI can perform these tasks. The policy must allow the
+    #   SageMaker AI service principal (sagemaker.amazonaws.com) permissions
+    #   to assume this role. For more information, see [SageMaker AI
+    #   Roles][1].
+    #
+    #   <note markdown="1"> To be able to pass this role to SageMaker AI, the caller of this API
     #   must have the `iam:PassRole` permission.
     #
     #    </note>
@@ -8790,7 +9161,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] kms_key_id
     #   The Amazon Resource Name (ARN) of a Amazon Web Services Key
-    #   Management Service key that SageMaker uses to encrypt data on the
+    #   Management Service key that SageMaker AI uses to encrypt data on the
     #   storage volume attached to your notebook instance. The KMS key you
     #   provide must be enabled. For information, see [Enabling and
     #   Disabling Keys][1] in the *Amazon Web Services Key Management
@@ -8823,10 +9194,10 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] direct_internet_access
-    #   Sets whether SageMaker provides internet access to the notebook
+    #   Sets whether SageMaker AI provides internet access to the notebook
     #   instance. If you set this to `Disabled` this notebook instance is
     #   able to access resources only in your VPC, and is not be able to
-    #   connect to SageMaker training and endpoint services unless you
+    #   connect to SageMaker AI training and endpoint services unless you
     #   configure a NAT Gateway in your VPC.
     #
     #   For more information, see [Notebook Instances Are Internet-Enabled
@@ -8858,7 +9229,8 @@ module Aws::SageMaker
     #   repository in [Amazon Web Services CodeCommit][1] or in any other
     #   Git repository. When you open a notebook instance, it opens in the
     #   directory that contains this repository. For more information, see
-    #   [Associating Git Repositories with SageMaker Notebook Instances][2].
+    #   [Associating Git Repositories with SageMaker AI Notebook
+    #   Instances][2].
     #
     #
     #
@@ -8873,7 +9245,7 @@ module Aws::SageMaker
     #   in [Amazon Web Services CodeCommit][1] or in any other Git
     #   repository. These repositories are cloned at the same level as the
     #   default repository of your notebook instance. For more information,
-    #   see [Associating Git Repositories with SageMaker Notebook
+    #   see [Associating Git Repositories with SageMaker AI Notebook
     #   Instances][2].
     #
     #
@@ -8981,9 +9353,9 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that enables Amazon
-    #   SageMaker to perform tasks on your behalf.
+    #   SageMaker AI to perform tasks on your behalf.
     #
-    #   During model optimization, Amazon SageMaker needs your permission
+    #   During model optimization, Amazon SageMaker AI needs your permission
     #   to:
     #
     #   * Read input data from an S3 bucket
@@ -8995,9 +9367,9 @@ module Aws::SageMaker
     #   * Publish metrics to Amazon CloudWatch
     #
     #   You grant permissions for all of these tasks to an IAM role. To pass
-    #   this role to Amazon SageMaker, the caller of this API must have the
-    #   `iam:PassRole` permission. For more information, see [Amazon
-    #   SageMaker Roles.][1]
+    #   this role to Amazon SageMaker AI, the caller of this API must have
+    #   the `iam:PassRole` permission. For more information, see [Amazon
+    #   SageMaker AI Roles.][1]
     #
     #
     #
@@ -9095,6 +9467,123 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App to create the presigned URL
+    #   for.
+    #   @return [String]
+    #
+    # @!attribute [rw] expires_in_seconds
+    #   The time that will pass before the presigned URL expires.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] session_expiration_duration_in_seconds
+    #   Indicates how long the Amazon SageMaker Partner AI App session can
+    #   be accessed for after logging in.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreatePartnerAppPresignedUrlRequest AWS API Documentation
+    #
+    class CreatePartnerAppPresignedUrlRequest < Struct.new(
+      :arn,
+      :expires_in_seconds,
+      :session_expiration_duration_in_seconds)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] url
+    #   The presigned URL that you can use to access the SageMaker Partner
+    #   AI App.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreatePartnerAppPresignedUrlResponse AWS API Documentation
+    #
+    class CreatePartnerAppPresignedUrlResponse < Struct.new(
+      :url)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] name
+    #   The name to give the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The type of SageMaker Partner AI App to create. Must be one of the
+    #   following: `lakera-guard`, `comet`, `deepchecks-llm-evaluation`, or
+    #   `fiddler`.
+    #   @return [String]
+    #
+    # @!attribute [rw] execution_role_arn
+    #   The ARN of the IAM role that the partner application uses.
+    #   @return [String]
+    #
+    # @!attribute [rw] maintenance_config
+    #   Maintenance configuration settings for the SageMaker Partner AI App.
+    #   @return [Types::PartnerAppMaintenanceConfig]
+    #
+    # @!attribute [rw] tier
+    #   Indicates the instance type and size of the cluster attached to the
+    #   SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] application_config
+    #   Configuration settings for the SageMaker Partner AI App.
+    #   @return [Types::PartnerAppConfig]
+    #
+    # @!attribute [rw] auth_type
+    #   The authorization type that users use to access the SageMaker
+    #   Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] enable_iam_session_based_identity
+    #   When set to `TRUE`, the SageMaker Partner AI App sets the Amazon Web
+    #   Services IAM session name or the authenticated IAM user as the
+    #   identity of the SageMaker Partner AI App user.
+    #   @return [Boolean]
+    #
+    # @!attribute [rw] client_token
+    #   A unique token that guarantees that the call to this API is
+    #   idempotent.
+    #
+    #   **A suitable default value is auto-generated.** You should normally
+    #   not need to pass this option.
+    #   @return [String]
+    #
+    # @!attribute [rw] tags
+    #   Each tag consists of a key and an optional value. Tag keys must be
+    #   unique per resource.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreatePartnerAppRequest AWS API Documentation
+    #
+    class CreatePartnerAppRequest < Struct.new(
+      :name,
+      :type,
+      :execution_role_arn,
+      :maintenance_config,
+      :tier,
+      :application_config,
+      :auth_type,
+      :enable_iam_session_based_identity,
+      :client_token,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreatePartnerAppResponse AWS API Documentation
+    #
+    class CreatePartnerAppResponse < Struct.new(
+      :arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] pipeline_name
     #   The name of the pipeline.
     #   @return [String]
@@ -9529,13 +10018,13 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] studio_lifecycle_config_name
-    #   The name of the Amazon SageMaker Studio Lifecycle Configuration to
-    #   create.
+    #   The name of the Amazon SageMaker AI Studio Lifecycle Configuration
+    #   to create.
     #   @return [String]
     #
     # @!attribute [rw] studio_lifecycle_config_content
-    #   The content of your Amazon SageMaker Studio Lifecycle Configuration
-    #   script. This content must be base64 encoded.
+    #   The content of your Amazon SageMaker AI Studio Lifecycle
+    #   Configuration script. This content must be base64 encoded.
     #   @return [String]
     #
     # @!attribute [rw] studio_lifecycle_config_app_type
@@ -9873,6 +10362,41 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_plan_name
+    #   The name of the training plan to create.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_plan_offering_id
+    #   The unique identifier of the training plan offering to use for
+    #   creating this plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] tags
+    #   An array of key-value pairs to apply to this training plan.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateTrainingPlanRequest AWS API Documentation
+    #
+    class CreateTrainingPlanRequest < Struct.new(
+      :training_plan_name,
+      :training_plan_offering_id,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the created training plan.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CreateTrainingPlanResponse AWS API Documentation
+    #
+    class CreateTrainingPlanResponse < Struct.new(
+      :training_plan_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] transform_job_name
     #   The name of the transform job. The name must be unique within an
     #   Amazon Web Services Region in an Amazon Web Services account.
@@ -10404,8 +10928,8 @@ module Aws::SageMaker
     end
 
     # A file system, created by you, that you assign to a user profile or
-    # space for an Amazon SageMaker Domain. Permitted users can access this
-    # file system in Amazon SageMaker Studio.
+    # space for an Amazon SageMaker AI Domain. Permitted users can access
+    # this file system in Amazon SageMaker AI Studio.
     #
     # @note CustomFileSystem is a union - when making an API calls you must set exactly one of the members.
     #
@@ -10415,22 +10939,28 @@ module Aws::SageMaker
     #   A custom file system in Amazon EFS.
     #   @return [Types::EFSFileSystem]
     #
+    # @!attribute [rw] f_sx_lustre_file_system
+    #   A custom file system in Amazon FSx for Lustre.
+    #   @return [Types::FSxLustreFileSystem]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CustomFileSystem AWS API Documentation
     #
     class CustomFileSystem < Struct.new(
       :efs_file_system,
+      :f_sx_lustre_file_system,
       :unknown)
       SENSITIVE = []
       include Aws::Structure
       include Aws::Structure::Union
 
       class EfsFileSystem < CustomFileSystem; end
+      class FSxLustreFileSystem < CustomFileSystem; end
       class Unknown < CustomFileSystem; end
     end
 
     # The settings for assigning a custom file system to a user profile or
-    # space for an Amazon SageMaker Domain. Permitted users can access this
-    # file system in Amazon SageMaker Studio.
+    # space for an Amazon SageMaker AI Domain. Permitted users can access
+    # this file system in Amazon SageMaker AI Studio.
     #
     # @note CustomFileSystemConfig is a union - when making an API calls you must set exactly one of the members.
     #
@@ -10440,21 +10970,27 @@ module Aws::SageMaker
     #   The settings for a custom Amazon EFS file system.
     #   @return [Types::EFSFileSystemConfig]
     #
+    # @!attribute [rw] f_sx_lustre_file_system_config
+    #   The settings for a custom Amazon FSx for Lustre file system.
+    #   @return [Types::FSxLustreFileSystemConfig]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/CustomFileSystemConfig AWS API Documentation
     #
     class CustomFileSystemConfig < Struct.new(
       :efs_file_system_config,
+      :f_sx_lustre_file_system_config,
       :unknown)
       SENSITIVE = []
       include Aws::Structure
       include Aws::Structure::Union
 
       class EfsFileSystemConfig < CustomFileSystemConfig; end
+      class FSxLustreFileSystemConfig < CustomFileSystemConfig; end
       class Unknown < CustomFileSystemConfig; end
     end
 
-    # A custom SageMaker image. For more information, see [Bring your own
-    # SageMaker image][1].
+    # A custom SageMaker AI image. For more information, see [Bring your own
+    # SageMaker AI image][1].
     #
     #
     #
@@ -10526,7 +11062,7 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # Configuration to control how SageMaker captures inference data.
+    # Configuration to control how SageMaker AI captures inference data.
     #
     # @!attribute [rw] enable_capture
     #   Whether data capture should be enabled or disabled (defaults to
@@ -10534,8 +11070,8 @@ module Aws::SageMaker
     #   @return [Boolean]
     #
     # @!attribute [rw] initial_sampling_percentage
-    #   The percentage of requests SageMaker will capture. A lower value is
-    #   recommended for Endpoints with high traffic.
+    #   The percentage of requests SageMaker AI will capture. A lower value
+    #   is recommended for Endpoints with high traffic.
     #   @return [Integer]
     #
     # @!attribute [rw] destination_s3_uri
@@ -10544,8 +11080,8 @@ module Aws::SageMaker
     #
     # @!attribute [rw] kms_key_id
     #   The Amazon Resource Name (ARN) of an Key Management Service key that
-    #   SageMaker uses to encrypt the captured data at rest using Amazon S3
-    #   server-side encryption.
+    #   SageMaker AI uses to encrypt the captured data at rest using Amazon
+    #   S3 server-side encryption.
     #
     #   The KmsKeyId can be any of the following formats:
     #
@@ -10567,8 +11103,8 @@ module Aws::SageMaker
     #
     # @!attribute [rw] capture_content_type_header
     #   Configuration specifying how to treat different headers. If no
-    #   headers are specified SageMaker will by default base64 encode when
-    #   capturing the data.
+    #   headers are specified SageMaker AI will by default base64 encode
+    #   when capturing the data.
     #   @return [Types::CaptureContentTypeHeader]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DataCaptureConfig AWS API Documentation
@@ -11078,7 +11614,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] custom_file_system_configs
     #   The settings for assigning a custom file system to a domain.
-    #   Permitted users can access this file system in Amazon SageMaker
+    #   Permitted users can access this file system in Amazon SageMaker AI
     #   Studio.
     #   @return [Array<Types::CustomFileSystemConfig>]
     #
@@ -11282,6 +11818,18 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] cluster_scheduler_config_id
+    #   ID of the cluster policy.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeleteClusterSchedulerConfigRequest AWS API Documentation
+    #
+    class DeleteClusterSchedulerConfigRequest < Struct.new(
+      :cluster_scheduler_config_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] code_repository_name
     #   The name of the Git repository to delete.
     #   @return [String]
@@ -11306,6 +11854,18 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] compute_quota_id
+    #   ID of the compute allocation definition.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeleteComputeQuotaRequest AWS API Documentation
+    #
+    class DeleteComputeQuotaRequest < Struct.new(
+      :compute_quota_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] context_name
     #   The name of the context to delete.
     #   @return [String]
@@ -11789,7 +12349,7 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] notebook_instance_name
-    #   The name of the SageMaker notebook instance to delete.
+    #   The name of the SageMaker AI notebook instance to delete.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeleteNotebookInstanceInput AWS API Documentation
@@ -11824,6 +12384,39 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App to delete.
+    #   @return [String]
+    #
+    # @!attribute [rw] client_token
+    #   A unique token that guarantees that the call to this API is
+    #   idempotent.
+    #
+    #   **A suitable default value is auto-generated.** You should normally
+    #   not need to pass this option.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeletePartnerAppRequest AWS API Documentation
+    #
+    class DeletePartnerAppRequest < Struct.new(
+      :arn,
+      :client_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App that was deleted.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeletePartnerAppResponse AWS API Documentation
+    #
+    class DeletePartnerAppResponse < Struct.new(
+      :arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] pipeline_name
     #   The name of the pipeline to delete.
     #   @return [String]
@@ -11888,8 +12481,8 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] studio_lifecycle_config_name
-    #   The name of the Amazon SageMaker Studio Lifecycle Configuration to
-    #   delete.
+    #   The name of the Amazon SageMaker AI Studio Lifecycle Configuration
+    #   to delete.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DeleteStudioLifecycleConfigRequest AWS API Documentation
@@ -12507,15 +13100,15 @@ module Aws::SageMaker
     #
     # @!attribute [rw] last_user_activity_timestamp
     #   The timestamp of the last user's activity.
-    #   `LastUserActivityTimestamp` is also updated when SageMaker performs
-    #   health checks without user activity. As a result, this value is set
-    #   to the same value as `LastHealthCheckTimestamp`.
+    #   `LastUserActivityTimestamp` is also updated when SageMaker AI
+    #   performs health checks without user activity. As a result, this
+    #   value is set to the same value as `LastHealthCheckTimestamp`.
     #   @return [Time]
     #
     # @!attribute [rw] creation_time
     #   The creation time of the application.
     #
-    #   <note markdown="1"> After an application has been shut down for 24 hours, SageMaker
+    #   <note markdown="1"> After an application has been shut down for 24 hours, SageMaker AI
     #   deletes all metadata for the application. To be considered an update
     #   and retain application metadata, applications must be restarted
     #   within 24 hours after the previous application has been shut down.
@@ -12531,7 +13124,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] resource_spec
     #   The instance type and the Amazon Resource Name (ARN) of the
-    #   SageMaker image created on the instance.
+    #   SageMaker AI image created on the instance.
     #   @return [Types::ResourceSpec]
     #
     # @!attribute [rw] built_in_lifecycle_config_arn
@@ -12703,9 +13296,9 @@ module Aws::SageMaker
     #   @return [Array<Types::AutoMLPartialFailureReason>]
     #
     # @!attribute [rw] best_candidate
-    #   The best model candidate selected by SageMaker Autopilot using both
-    #   the best objective metric and lowest [InferenceLatency][1] for an
-    #   experiment.
+    #   The best model candidate selected by SageMaker AI Autopilot using
+    #   both the best objective metric and lowest [InferenceLatency][1] for
+    #   an experiment.
     #
     #
     #
@@ -13022,6 +13615,99 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] cluster_scheduler_config_id
+    #   ID of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_scheduler_config_version
+    #   Version of the cluster policy.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterSchedulerConfigRequest AWS API Documentation
+    #
+    class DescribeClusterSchedulerConfigRequest < Struct.new(
+      :cluster_scheduler_config_id,
+      :cluster_scheduler_config_version)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] cluster_scheduler_config_arn
+    #   ARN of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_scheduler_config_id
+    #   ID of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] name
+    #   Name of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_scheduler_config_version
+    #   Version of the cluster policy.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] status
+    #   Status of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] failure_reason
+    #   Failure reason of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   ARN of the cluster where the cluster policy is applied.
+    #   @return [String]
+    #
+    # @!attribute [rw] scheduler_config
+    #   Cluster policy configuration. This policy is used for task
+    #   prioritization and fair-share allocation. This helps prioritize
+    #   critical workloads and distributes idle compute across entities.
+    #   @return [Types::SchedulerConfig]
+    #
+    # @!attribute [rw] description
+    #   Description of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   Creation time of the cluster policy.
+    #   @return [Time]
+    #
+    # @!attribute [rw] created_by
+    #   Information about the user who created or modified an experiment,
+    #   trial, trial component, lineage group, project, or model card.
+    #   @return [Types::UserContext]
+    #
+    # @!attribute [rw] last_modified_time
+    #   Last modified time of the cluster policy.
+    #   @return [Time]
+    #
+    # @!attribute [rw] last_modified_by
+    #   Information about the user who created or modified an experiment,
+    #   trial, trial component, lineage group, project, or model card.
+    #   @return [Types::UserContext]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeClusterSchedulerConfigResponse AWS API Documentation
+    #
+    class DescribeClusterSchedulerConfigResponse < Struct.new(
+      :cluster_scheduler_config_arn,
+      :cluster_scheduler_config_id,
+      :name,
+      :cluster_scheduler_config_version,
+      :status,
+      :failure_reason,
+      :cluster_arn,
+      :scheduler_config,
+      :description,
+      :creation_time,
+      :created_by,
+      :last_modified_time,
+      :last_modified_by)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] code_repository_name
     #   The name of the Git repository to describe.
     #   @return [String]
@@ -13109,12 +13795,12 @@ module Aws::SageMaker
     #   The time when the model compilation job on a compilation job
     #   instance ended. For a successful or stopped job, this is when the
     #   job's model artifacts have finished uploading. For a failed job,
-    #   this is when Amazon SageMaker detected that the job failed.
+    #   this is when Amazon SageMaker AI detected that the job failed.
     #   @return [Time]
     #
     # @!attribute [rw] stopping_condition
     #   Specifies a limit to how long a model compilation job can run. When
-    #   the job reaches the time limit, Amazon SageMaker ends the
+    #   the job reaches the time limit, Amazon SageMaker AI ends the
     #   compilation job. Use this API to cap model training costs.
     #   @return [Types::StoppingCondition]
     #
@@ -13153,7 +13839,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   assumes to perform the model compilation job.
+    #   AI assumes to perform the model compilation job.
     #   @return [String]
     #
     # @!attribute [rw] input_config
@@ -13209,6 +13895,112 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] compute_quota_id
+    #   ID of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_version
+    #   Version of the compute allocation definition.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeComputeQuotaRequest AWS API Documentation
+    #
+    class DescribeComputeQuotaRequest < Struct.new(
+      :compute_quota_id,
+      :compute_quota_version)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] compute_quota_arn
+    #   ARN of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_id
+    #   ID of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] name
+    #   Name of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] description
+    #   Description of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_version
+    #   Version of the compute allocation definition.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] status
+    #   Status of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] failure_reason
+    #   Failure reason of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   ARN of the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_config
+    #   Configuration of the compute allocation definition. This includes
+    #   the resource sharing option, and the setting to preempt low priority
+    #   tasks.
+    #   @return [Types::ComputeQuotaConfig]
+    #
+    # @!attribute [rw] compute_quota_target
+    #   The target entity to allocate compute resources to.
+    #   @return [Types::ComputeQuotaTarget]
+    #
+    # @!attribute [rw] activation_state
+    #   The state of the compute allocation being described. Use to enable
+    #   or disable compute allocation.
+    #
+    #   Default is `Enabled`.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   Creation time of the compute allocation configuration.
+    #   @return [Time]
+    #
+    # @!attribute [rw] created_by
+    #   Information about the user who created or modified an experiment,
+    #   trial, trial component, lineage group, project, or model card.
+    #   @return [Types::UserContext]
+    #
+    # @!attribute [rw] last_modified_time
+    #   Last modified time of the compute allocation configuration.
+    #   @return [Time]
+    #
+    # @!attribute [rw] last_modified_by
+    #   Information about the user who created or modified an experiment,
+    #   trial, trial component, lineage group, project, or model card.
+    #   @return [Types::UserContext]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeComputeQuotaResponse AWS API Documentation
+    #
+    class DescribeComputeQuotaResponse < Struct.new(
+      :compute_quota_arn,
+      :compute_quota_id,
+      :name,
+      :description,
+      :compute_quota_version,
+      :status,
+      :failure_reason,
+      :cluster_arn,
+      :compute_quota_config,
+      :compute_quota_target,
+      :activation_state,
+      :creation_time,
+      :created_by,
+      :last_modified_time,
+      :last_modified_by)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] context_name
     #   The name of the context to describe.
     #   @return [String]
@@ -13340,7 +14132,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform tasks on your behalf.
+    #   AI can assume to perform tasks on your behalf.
     #   @return [String]
     #
     # @!attribute [rw] stopping_condition
@@ -13546,7 +14338,7 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] single_sign_on_application_arn
-    #   The ARN of the application managed by SageMaker in IAM Identity
+    #   The ARN of the application managed by SageMaker AI in IAM Identity
     #   Center. This value is only returned for domains created after
     #   October 1, 2023.
     #   @return [String]
@@ -13590,7 +14382,7 @@ module Aws::SageMaker
     #   `PublicInternetOnly`.
     #
     #   * `PublicInternetOnly` - Non-EFS traffic is through a VPC managed by
-    #     Amazon SageMaker, which allows direct internet access
+    #     Amazon SageMaker AI, which allows direct internet access
     #
     #   * `VpcOnly` - All traffic is through the specified VPC and subnets
     #   @return [String]
@@ -13876,7 +14668,7 @@ module Aws::SageMaker
     #   @return [Array<Types::ProductionVariant>]
     #
     # @!attribute [rw] data_capture_config
-    #   Configuration to control how SageMaker captures inference data.
+    #   Configuration to control how SageMaker AI captures inference data.
     #   @return [Types::DataCaptureConfig]
     #
     # @!attribute [rw] kms_key_id
@@ -14934,7 +15726,7 @@ module Aws::SageMaker
     #   @return [Time]
     #
     # @!attribute [rw] role_arn
-    #   The ARN of the IAM role that enables Amazon SageMaker to perform
+    #   The ARN of the IAM role that enables Amazon SageMaker AI to perform
     #   tasks on your behalf.
     #   @return [String]
     #
@@ -15032,16 +15824,16 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] job_type
-    #   Indicates SageMaker job type compatibility.
+    #   Indicates SageMaker AI job type compatibility.
     #
-    #   * `TRAINING`: The image version is compatible with SageMaker
+    #   * `TRAINING`: The image version is compatible with SageMaker AI
     #     training jobs.
     #
-    #   * `INFERENCE`: The image version is compatible with SageMaker
+    #   * `INFERENCE`: The image version is compatible with SageMaker AI
     #     inference jobs.
     #
     #   * `NOTEBOOK_KERNEL`: The image version is compatible with SageMaker
-    #     notebook kernels.
+    #     AI notebook kernels.
     #   @return [String]
     #
     # @!attribute [rw] ml_framework
@@ -16493,7 +17285,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform tasks on your behalf.
+    #   AI can assume to perform tasks on your behalf.
     #   @return [String]
     #
     # @!attribute [rw] stopping_condition
@@ -16671,7 +17463,7 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] notebook_instance_name
-    #   The name of the SageMaker notebook instance.
+    #   The name of the SageMaker AI notebook instance.
     #   @return [String]
     #
     # @!attribute [rw] notebook_instance_status
@@ -16705,12 +17497,12 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] kms_key_id
-    #   The Amazon Web Services KMS key ID SageMaker uses to encrypt data
+    #   The Amazon Web Services KMS key ID SageMaker AI uses to encrypt data
     #   when storing it on the ML storage volume attached to the instance.
     #   @return [String]
     #
     # @!attribute [rw] network_interface_id
-    #   The network interface IDs that SageMaker created at the time of
+    #   The network interface IDs that SageMaker AI created at the time of
     #   creating the instance.
     #   @return [String]
     #
@@ -16736,10 +17528,10 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] direct_internet_access
-    #   Describes whether SageMaker provides internet access to the notebook
-    #   instance. If this value is set to *Disabled*, the notebook instance
-    #   does not have internet access, and cannot connect to SageMaker
-    #   training and endpoint services.
+    #   Describes whether SageMaker AI provides internet access to the
+    #   notebook instance. If this value is set to *Disabled*, the notebook
+    #   instance does not have internet access, and cannot connect to
+    #   SageMaker AI training and endpoint services.
     #
     #   For more information, see [Notebook Instances Are Internet-Enabled
     #   by Default][1].
@@ -16769,7 +17561,8 @@ module Aws::SageMaker
     #   repository in [Amazon Web Services CodeCommit][1] or in any other
     #   Git repository. When you open a notebook instance, it opens in the
     #   directory that contains this repository. For more information, see
-    #   [Associating Git Repositories with SageMaker Notebook Instances][2].
+    #   [Associating Git Repositories with SageMaker AI Notebook
+    #   Instances][2].
     #
     #
     #
@@ -16784,7 +17577,7 @@ module Aws::SageMaker
     #   in [Amazon Web Services CodeCommit][1] or in any other Git
     #   repository. These repositories are cloned at the same level as the
     #   default repository of your notebook instance. For more information,
-    #   see [Associating Git Repositories with SageMaker Notebook
+    #   see [Associating Git Repositories with SageMaker AI Notebook
     #   Instances][2].
     #
     #
@@ -16972,6 +17765,103 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App to describe.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribePartnerAppRequest AWS API Documentation
+    #
+    class DescribePartnerAppRequest < Struct.new(
+      :arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App that was described.
+    #   @return [String]
+    #
+    # @!attribute [rw] name
+    #   The name of the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The type of SageMaker Partner AI App. Must be one of the following:
+    #   `lakera-guard`, `comet`, `deepchecks-llm-evaluation`, or `fiddler`.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The status of the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   The time that the SageMaker Partner AI App was created.
+    #   @return [Time]
+    #
+    # @!attribute [rw] execution_role_arn
+    #   The ARN of the IAM role associated with the SageMaker Partner AI
+    #   App.
+    #   @return [String]
+    #
+    # @!attribute [rw] base_url
+    #   The URL of the SageMaker Partner AI App that the Application SDK
+    #   uses to support in-app calls for the user.
+    #   @return [String]
+    #
+    # @!attribute [rw] maintenance_config
+    #   Maintenance configuration settings for the SageMaker Partner AI App.
+    #   @return [Types::PartnerAppMaintenanceConfig]
+    #
+    # @!attribute [rw] tier
+    #   The instance type and size of the cluster attached to the SageMaker
+    #   Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] version
+    #   The version of the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] application_config
+    #   Configuration settings for the SageMaker Partner AI App.
+    #   @return [Types::PartnerAppConfig]
+    #
+    # @!attribute [rw] auth_type
+    #   The authorization type that users use to access the SageMaker
+    #   Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] enable_iam_session_based_identity
+    #   When set to `TRUE`, the SageMaker Partner AI App sets the Amazon Web
+    #   Services IAM session name or the authenticated IAM user as the
+    #   identity of the SageMaker Partner AI App user.
+    #   @return [Boolean]
+    #
+    # @!attribute [rw] error
+    #   This is an error field object that contains the error code and the
+    #   reason for an operation failure.
+    #   @return [Types::ErrorInfo]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribePartnerAppResponse AWS API Documentation
+    #
+    class DescribePartnerAppResponse < Struct.new(
+      :arn,
+      :name,
+      :type,
+      :status,
+      :creation_time,
+      :execution_role_arn,
+      :base_url,
+      :maintenance_config,
+      :tier,
+      :version,
+      :application_config,
+      :auth_type,
+      :enable_iam_session_based_identity,
+      :error)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] pipeline_execution_arn
     #   The Amazon Resource Name (ARN) of the pipeline execution.
     #   @return [String]
@@ -17491,8 +18381,8 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] studio_lifecycle_config_name
-    #   The name of the Amazon SageMaker Studio Lifecycle Configuration to
-    #   describe.
+    #   The name of the Amazon SageMaker AI Studio Lifecycle Configuration
+    #   to describe.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeStudioLifecycleConfigRequest AWS API Documentation
@@ -17508,23 +18398,23 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] studio_lifecycle_config_name
-    #   The name of the Amazon SageMaker Studio Lifecycle Configuration that
-    #   is described.
+    #   The name of the Amazon SageMaker AI Studio Lifecycle Configuration
+    #   that is described.
     #   @return [String]
     #
     # @!attribute [rw] creation_time
-    #   The creation time of the Amazon SageMaker Studio Lifecycle
+    #   The creation time of the Amazon SageMaker AI Studio Lifecycle
     #   Configuration.
     #   @return [Time]
     #
     # @!attribute [rw] last_modified_time
-    #   This value is equivalent to CreationTime because Amazon SageMaker
+    #   This value is equivalent to CreationTime because Amazon SageMaker AI
     #   Studio Lifecycle Configurations are immutable.
     #   @return [Time]
     #
     # @!attribute [rw] studio_lifecycle_config_content
-    #   The content of your Amazon SageMaker Studio Lifecycle Configuration
-    #   script.
+    #   The content of your Amazon SageMaker AI Studio Lifecycle
+    #   Configuration script.
     #   @return [String]
     #
     # @!attribute [rw] studio_lifecycle_config_app_type
@@ -17976,6 +18866,117 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] training_plan_name
+    #   The name of the training plan to describe.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeTrainingPlanRequest AWS API Documentation
+    #
+    class DescribeTrainingPlanRequest < Struct.new(
+      :training_plan_name)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_plan_name
+    #   The name of the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The current status of the training plan (e.g., Pending, Active,
+    #   Expired). To see the complete list of status values available for a
+    #   training plan, refer to the `Status` attribute within the `
+    #   TrainingPlanSummary ` object.
+    #   @return [String]
+    #
+    # @!attribute [rw] status_message
+    #   A message providing additional information about the current status
+    #   of the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] duration_hours
+    #   The number of whole hours in the total duration for this training
+    #   plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] duration_minutes
+    #   The additional minutes beyond whole hours in the total duration for
+    #   this training plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] start_time
+    #   The start time of the training plan.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time
+    #   The end time of the training plan.
+    #   @return [Time]
+    #
+    # @!attribute [rw] upfront_fee
+    #   The upfront fee for the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] currency_code
+    #   The currency code for the upfront fee (e.g., USD).
+    #   @return [String]
+    #
+    # @!attribute [rw] total_instance_count
+    #   The total number of instances reserved in this training plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] available_instance_count
+    #   The number of instances currently available for use in this training
+    #   plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] in_use_instance_count
+    #   The number of instances currently in use from 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.
+    #
+    #   Training plans are specific to their target resource.
+    #
+    #   * A training plan designed for SageMaker training jobs can only be
+    #     used to schedule and run training jobs.
+    #
+    #   * A training plan for HyperPod clusters can be used exclusively to
+    #     provide compute resources to a cluster's instance group.
+    #   @return [Array<String>]
+    #
+    # @!attribute [rw] reserved_capacity_summaries
+    #   The list of Reserved Capacity providing the underlying compute
+    #   resources of the plan.
+    #   @return [Array<Types::ReservedCapacitySummary>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/DescribeTrainingPlanResponse AWS API Documentation
+    #
+    class DescribeTrainingPlanResponse < Struct.new(
+      :training_plan_arn,
+      :training_plan_name,
+      :status,
+      :status_message,
+      :duration_hours,
+      :duration_minutes,
+      :start_time,
+      :end_time,
+      :upfront_fee,
+      :currency_code,
+      :total_instance_count,
+      :available_instance_count,
+      :in_use_instance_count,
+      :target_resources,
+      :reserved_capacity_summaries)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] transform_job_name
     #   The name of the transform job that you want to view details of.
     #   @return [String]
@@ -18878,8 +19879,8 @@ module Aws::SageMaker
     #   @return [Types::RStudioServerProDomainSettings]
     #
     # @!attribute [rw] execution_role_identity_config
-    #   The configuration for attaching a SageMaker user profile name to the
-    #   execution role as a [sts:SourceIdentity key][1].
+    #   The configuration for attaching a SageMaker AI user profile name to
+    #   the execution role as a [sts:SourceIdentity key][1].
     #
     #
     #
@@ -18918,10 +19919,10 @@ module Aws::SageMaker
     #   @return [Types::RStudioServerProDomainSettingsForUpdate]
     #
     # @!attribute [rw] execution_role_identity_config
-    #   The configuration for attaching a SageMaker user profile name to the
-    #   execution role as a [sts:SourceIdentity key][1]. This configuration
-    #   can only be modified if there are no apps in the `InService` or
-    #   `Pending` state.
+    #   The configuration for attaching a SageMaker AI user profile name to
+    #   the execution role as a [sts:SourceIdentity key][1]. This
+    #   configuration can only be modified if there are no apps in the
+    #   `InService` or `Pending` state.
     #
     #
     #
@@ -19113,8 +20114,8 @@ module Aws::SageMaker
     end
 
     # A file system, created by you in Amazon EFS, that you assign to a user
-    # profile or space for an Amazon SageMaker Domain. Permitted users can
-    # access this file system in Amazon SageMaker Studio.
+    # profile or space for an Amazon SageMaker AI Domain. Permitted users
+    # can access this file system in Amazon SageMaker AI Studio.
     #
     # @!attribute [rw] file_system_id
     #   The ID of your Amazon EFS file system.
@@ -19129,7 +20130,7 @@ module Aws::SageMaker
     end
 
     # The settings for assigning a custom Amazon EFS file system to a user
-    # profile or space for an Amazon SageMaker Domain.
+    # profile or space for an Amazon SageMaker AI Domain.
     #
     # @!attribute [rw] file_system_id
     #   The ID of your Amazon EFS file system.
@@ -19137,8 +20138,8 @@ module Aws::SageMaker
     #
     # @!attribute [rw] file_system_path
     #   The path to the file system directory that is accessible in Amazon
-    #   SageMaker Studio. Permitted users can access only this directory and
-    #   below.
+    #   SageMaker AI Studio. Permitted users can access only this directory
+    #   and below.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/EFSFileSystemConfig AWS API Documentation
@@ -20174,6 +21175,26 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # This is an error field object that contains the error code and the
+    # reason for an operation failure.
+    #
+    # @!attribute [rw] code
+    #   The error code for an invalid or failed operation.
+    #   @return [String]
+    #
+    # @!attribute [rw] reason
+    #   The failure reason for the operation.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ErrorInfo AWS API Documentation
+    #
+    class ErrorInfo < Struct.new(
+      :code,
+      :reason)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The properties of an experiment as returned by the [Search][1] API.
     #
     #
@@ -20384,6 +21405,43 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # A custom file system in Amazon FSx for Lustre.
+    #
+    # @!attribute [rw] file_system_id
+    #   Amazon FSx for Lustre file system ID.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/FSxLustreFileSystem AWS API Documentation
+    #
+    class FSxLustreFileSystem < Struct.new(
+      :file_system_id)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The settings for assigning a custom Amazon FSx for Lustre file system
+    # to a user profile or space for an Amazon SageMaker Domain.
+    #
+    # @!attribute [rw] file_system_id
+    #   The globally unique, 17-digit, ID of the file system, assigned by
+    #   Amazon FSx for Lustre.
+    #   @return [String]
+    #
+    # @!attribute [rw] file_system_path
+    #   The path to the file system directory that is accessible in Amazon
+    #   SageMaker Studio. Permitted users can access only this directory and
+    #   below.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/FSxLustreFileSystemConfig AWS API Documentation
+    #
+    class FSxLustreFileSystemConfig < Struct.new(
+      :file_system_id,
+      :file_system_path)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The container for the metadata for Fail step.
     #
     # @!attribute [rw] error_message
@@ -20710,7 +21768,7 @@ module Aws::SageMaker
     end
 
     # The Amazon Elastic File System storage configuration for a SageMaker
-    # image.
+    # AI image.
     #
     # @!attribute [rw] mount_path
     #   The path within the image to mount the user's EFS home directory.
@@ -24009,9 +25067,10 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # A SageMaker image. A SageMaker image represents a set of container
-    # images that are derived from a common base container image. Each of
-    # these container images is represented by a SageMaker `ImageVersion`.
+    # A SageMaker AI image. A SageMaker AI image represents a set of
+    # container images that are derived from a common base container image.
+    # Each of these container images is represented by a SageMaker AI
+    # `ImageVersion`.
     #
     # @!attribute [rw] creation_time
     #   When the image was created.
@@ -24107,7 +25166,7 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # A version of a SageMaker `Image`. A version represents an existing
+    # A version of a SageMaker AI `Image`. A version represents an existing
     # container image.
     #
     # @!attribute [rw] creation_time
@@ -24380,8 +25439,8 @@ module Aws::SageMaker
     # including the model, container, and compute resources.
     #
     # @!attribute [rw] model_name
-    #   The name of an existing SageMaker model object in your account that
-    #   you want to deploy with the inference component.
+    #   The name of an existing SageMaker AI model object in your account
+    #   that you want to deploy with the inference component.
     #   @return [String]
     #
     # @!attribute [rw] container
@@ -24441,7 +25500,7 @@ module Aws::SageMaker
     # component.
     #
     # @!attribute [rw] model_name
-    #   The name of the SageMaker model object that is deployed with the
+    #   The name of the SageMaker AI model object that is deployed with the
     #   inference component.
     #   @return [String]
     #
@@ -24591,8 +25650,8 @@ module Aws::SageMaker
     #
     # @!attribute [rw] content_type
     #   Configuration specifying how to treat different headers. If no
-    #   headers are specified Amazon SageMaker will by default base64 encode
-    #   when capturing the data.
+    #   headers are specified Amazon SageMaker AI will by default base64
+    #   encode when capturing the data.
     #   @return [Types::CaptureContentTypeHeader]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/InferenceExperimentDataStorageConfig AWS API Documentation
@@ -25282,13 +26341,13 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # The configuration for the file system and kernels in a SageMaker image
-    # running as a JupyterLab app. The `FileSystemConfig` object is not
-    # supported.
+    # The configuration for the file system and kernels in a SageMaker AI
+    # image running as a JupyterLab app. The `FileSystemConfig` object is
+    # not supported.
     #
     # @!attribute [rw] file_system_config
     #   The Amazon Elastic File System storage configuration for a SageMaker
-    #   image.
+    #   AI image.
     #   @return [Types::FileSystemConfig]
     #
     # @!attribute [rw] container_config
@@ -25307,7 +26366,7 @@ module Aws::SageMaker
     # The settings for the JupyterLab application.
     #
     # @!attribute [rw] default_resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
@@ -25364,7 +26423,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] default_resource_spec
     #   The default instance type and the Amazon Resource Name (ARN) of the
-    #   default SageMaker image used by the JupyterServer app. If you use
+    #   default SageMaker AI image used by the JupyterServer app. If you use
     #   the `LifecycleConfigArns` parameter, then this parameter is also
     #   required.
     #   @return [Types::ResourceSpec]
@@ -25381,8 +26440,8 @@ module Aws::SageMaker
     #   @return [Array<String>]
     #
     # @!attribute [rw] code_repositories
-    #   A list of Git repositories that SageMaker automatically displays to
-    #   users for cloning in the JupyterServer application.
+    #   A list of Git repositories that SageMaker AI automatically displays
+    #   to users for cloning in the JupyterServer application.
     #   @return [Array<Types::CodeRepository>]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/JupyterServerAppSettings AWS API Documentation
@@ -25415,9 +26474,9 @@ module Aws::SageMaker
     #
     # @!attribute [rw] default_resource_spec
     #   The default instance type and the Amazon Resource Name (ARN) of the
-    #   default SageMaker image used by the KernelGateway app.
+    #   default SageMaker AI image used by the KernelGateway app.
     #
-    #   <note markdown="1"> The Amazon SageMaker Studio UI does not use the default instance
+    #   <note markdown="1"> The Amazon SageMaker AI Studio UI does not use the default instance
     #   type value set here. The default instance type set here is used when
     #   Apps are created using the CLI or CloudFormation and the instance
     #   type parameter value is not passed.
@@ -25426,7 +26485,7 @@ module Aws::SageMaker
     #   @return [Types::ResourceSpec]
     #
     # @!attribute [rw] custom_images
-    #   A list of custom SageMaker images that are configured to run as a
+    #   A list of custom SageMaker AI images that are configured to run as a
     #   KernelGateway app.
     #   @return [Array<Types::CustomImage>]
     #
@@ -25450,8 +26509,8 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # The configuration for the file system and kernels in a SageMaker image
-    # running as a KernelGateway app.
+    # The configuration for the file system and kernels in a SageMaker AI
+    # image running as a KernelGateway app.
     #
     # @!attribute [rw] kernel_specs
     #   The specification of the Jupyter kernels in the image.
@@ -25459,7 +26518,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] file_system_config
     #   The Amazon Elastic File System storage configuration for a SageMaker
-    #   image.
+    #   AI image.
     #   @return [Types::FileSystemConfig]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/KernelGatewayImageConfig AWS API Documentation
@@ -26247,7 +27306,7 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] sage_maker_image_version_aliases
-    #   A list of SageMaker image version aliases.
+    #   A list of SageMaker AI image version aliases.
     #   @return [Array<String>]
     #
     # @!attribute [rw] next_token
@@ -26814,6 +27873,92 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] created_after
+    #   Filter for after this creation time. The input for this parameter is
+    #   a Unix timestamp. To convert a date and time into a Unix timestamp,
+    #   see [EpochConverter][1].
+    #
+    #
+    #
+    #   [1]: https://www.epochconverter.com/
+    #   @return [Time]
+    #
+    # @!attribute [rw] created_before
+    #   Filter for before this creation time. The input for this parameter
+    #   is a Unix timestamp. To convert a date and time into a Unix
+    #   timestamp, see [EpochConverter][1].
+    #
+    #
+    #
+    #   [1]: https://www.epochconverter.com/
+    #   @return [Time]
+    #
+    # @!attribute [rw] name_contains
+    #   Filter for name containing this string.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   Filter for ARN of the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   Filter for status.
+    #   @return [String]
+    #
+    # @!attribute [rw] sort_by
+    #   Filter for sorting the list by a given value. For example, sort by
+    #   name, creation time, or status.
+    #   @return [String]
+    #
+    # @!attribute [rw] sort_order
+    #   The order of the list. By default, listed in `Descending` order
+    #   according to by `SortBy`. To change the list order, you can specify
+    #   `SortOrder` to be `Ascending`.
+    #   @return [String]
+    #
+    # @!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.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   The maximum number of cluster policies to list.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListClusterSchedulerConfigsRequest AWS API Documentation
+    #
+    class ListClusterSchedulerConfigsRequest < Struct.new(
+      :created_after,
+      :created_before,
+      :name_contains,
+      :cluster_arn,
+      :status,
+      :sort_by,
+      :sort_order,
+      :next_token,
+      :max_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] cluster_scheduler_config_summaries
+    #   Summaries of the cluster policies.
+    #   @return [Array<Types::ClusterSchedulerConfigSummary>]
+    #
+    # @!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.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListClusterSchedulerConfigsResponse AWS API Documentation
+    #
+    class ListClusterSchedulerConfigsResponse < Struct.new(
+      :cluster_scheduler_config_summaries,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] creation_time_after
     #   Set a start time for the time range during which you want to list
     #   SageMaker HyperPod clusters. Timestamps are formatted according to
@@ -26877,6 +28022,13 @@ module Aws::SageMaker
     #   The sort order for results. The default value is `Ascending`.
     #   @return [String]
     #
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan to filter
+    #   clusters by. For more information about reserving GPU capacity for
+    #   your SageMaker HyperPod clusters using Amazon SageMaker Training
+    #   Plan, see ` CreateTrainingPlan `.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListClustersRequest AWS API Documentation
     #
     class ListClustersRequest < Struct.new(
@@ -26886,7 +28038,8 @@ module Aws::SageMaker
       :name_contains,
       :next_token,
       :sort_by,
-      :sort_order)
+      :sort_order,
+      :training_plan_arn)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -27078,7 +28231,7 @@ module Aws::SageMaker
     #   @return [Array<Types::CompilationJobSummary>]
     #
     # @!attribute [rw] next_token
-    #   If the response is truncated, Amazon SageMaker returns this
+    #   If the response is truncated, Amazon SageMaker AI returns this
     #   `NextToken`. To retrieve the next set of model compilation jobs, use
     #   this token in the next request.
     #   @return [String]
@@ -27092,6 +28245,92 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] created_after
+    #   Filter for after this creation time. The input for this parameter is
+    #   a Unix timestamp. To convert a date and time into a Unix timestamp,
+    #   see [EpochConverter][1].
+    #
+    #
+    #
+    #   [1]: https://www.epochconverter.com/
+    #   @return [Time]
+    #
+    # @!attribute [rw] created_before
+    #   Filter for before this creation time. The input for this parameter
+    #   is a Unix timestamp. To convert a date and time into a Unix
+    #   timestamp, see [EpochConverter][1].
+    #
+    #
+    #
+    #   [1]: https://www.epochconverter.com/
+    #   @return [Time]
+    #
+    # @!attribute [rw] name_contains
+    #   Filter for name containing this string.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   Filter for status.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_arn
+    #   Filter for ARN of the cluster.
+    #   @return [String]
+    #
+    # @!attribute [rw] sort_by
+    #   Filter for sorting the list by a given value. For example, sort by
+    #   name, creation time, or status.
+    #   @return [String]
+    #
+    # @!attribute [rw] sort_order
+    #   The order of the list. By default, listed in `Descending` order
+    #   according to by `SortBy`. To change the list order, you can specify
+    #   `SortOrder` to be `Ascending`.
+    #   @return [String]
+    #
+    # @!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.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   The maximum number of compute allocation definitions to list.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListComputeQuotasRequest AWS API Documentation
+    #
+    class ListComputeQuotasRequest < Struct.new(
+      :created_after,
+      :created_before,
+      :name_contains,
+      :status,
+      :cluster_arn,
+      :sort_by,
+      :sort_order,
+      :next_token,
+      :max_results)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] compute_quota_summaries
+    #   Summaries of the compute allocation definitions.
+    #   @return [Array<Types::ComputeQuotaSummary>]
+    #
+    # @!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.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListComputeQuotasResponse AWS API Documentation
+    #
+    class ListComputeQuotasResponse < Struct.new(
+      :compute_quota_summaries,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] source_uri
     #   A filter that returns only contexts with the specified source URI.
     #   @return [String]
@@ -29734,8 +30973,8 @@ module Aws::SageMaker
     #   @return [Array<Types::MonitoringJobDefinitionSummary>]
     #
     # @!attribute [rw] next_token
-    #   If the response is truncated, Amazon SageMaker returns this token.
-    #   To retrieve the next set of model quality monitoring job
+    #   If the response is truncated, Amazon SageMaker AI returns this
+    #   token. To retrieve the next set of model quality monitoring job
     #   definitions, use it in the next request.
     #   @return [String]
     #
@@ -30203,8 +31442,8 @@ module Aws::SageMaker
     end
 
     # @!attribute [rw] next_token
-    #   If the response is truncated, SageMaker returns this token. To get
-    #   the next set of lifecycle configurations, use it in the next
+    #   If the response is truncated, SageMaker AI returns this token. To
+    #   get the next set of lifecycle configurations, use it in the next
     #   request.
     #   @return [String]
     #
@@ -30318,8 +31557,8 @@ module Aws::SageMaker
 
     # @!attribute [rw] next_token
     #   If the response to the previous `ListNotebookInstances` request was
-    #   truncated, SageMaker returns this token. To retrieve the next set of
-    #   notebook instances, use the token in the next request.
+    #   truncated, SageMaker AI returns this token. To retrieve the next set
+    #   of notebook instances, use the token in the next request.
     #   @return [String]
     #
     # @!attribute [rw] notebook_instances
@@ -30429,6 +31668,49 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] max_results
+    #   This parameter defines the maximum number of results that can be
+    #   returned in a single response. The `MaxResults` parameter is an
+    #   upper bound, not a target. If there are more results available than
+    #   the value specified, a `NextToken` is provided in the response. The
+    #   `NextToken` indicates that the user should get the next set of
+    #   results by providing this token as a part of a subsequent call. The
+    #   default value for `MaxResults` is 10.
+    #   @return [Integer]
+    #
+    # @!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.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListPartnerAppsRequest AWS API Documentation
+    #
+    class ListPartnerAppsRequest < Struct.new(
+      :max_results,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] summaries
+    #   The information related to each of the SageMaker Partner AI Apps in
+    #   an account.
+    #   @return [Array<Types::PartnerAppSummary>]
+    #
+    # @!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.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListPartnerAppsResponse AWS API Documentation
+    #
+    class ListPartnerAppsResponse < Struct.new(
+      :summaries,
+      :next_token)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] pipeline_execution_arn
     #   The Amazon Resource Name (ARN) of the pipeline execution.
     #   @return [String]
@@ -31291,6 +32573,13 @@ module Aws::SageMaker
     #   status.
     #   @return [String]
     #
+    # @!attribute [rw] training_plan_arn_equals
+    #   The Amazon Resource Name (ARN); of the training plan to filter
+    #   training jobs by. For more information about reserving GPU capacity
+    #   for your SageMaker training jobs using Amazon SageMaker Training
+    #   Plan, see ` CreateTrainingPlan `.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListTrainingJobsRequest AWS API Documentation
     #
     class ListTrainingJobsRequest < Struct.new(
@@ -31304,7 +32593,8 @@ module Aws::SageMaker
       :status_equals,
       :sort_by,
       :sort_order,
-      :warm_pool_status_equals)
+      :warm_pool_status_equals,
+      :training_plan_arn_equals)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -31329,6 +32619,68 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] next_token
+    #   A token to continue pagination if more results are available.
+    #   @return [String]
+    #
+    # @!attribute [rw] max_results
+    #   The maximum number of results to return in the response.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] start_time_after
+    #   Filter to list only training plans with an actual start time after
+    #   this date.
+    #   @return [Time]
+    #
+    # @!attribute [rw] start_time_before
+    #   Filter to list only training plans with an actual start time before
+    #   this date.
+    #   @return [Time]
+    #
+    # @!attribute [rw] sort_by
+    #   The training plan field to sort the results by (e.g., StartTime,
+    #   Status).
+    #   @return [String]
+    #
+    # @!attribute [rw] sort_order
+    #   The order to sort the results (Ascending or Descending).
+    #   @return [String]
+    #
+    # @!attribute [rw] filters
+    #   Additional filters to apply to the list of training plans.
+    #   @return [Array<Types::TrainingPlanFilter>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListTrainingPlansRequest AWS API Documentation
+    #
+    class ListTrainingPlansRequest < Struct.new(
+      :next_token,
+      :max_results,
+      :start_time_after,
+      :start_time_before,
+      :sort_by,
+      :sort_order,
+      :filters)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] next_token
+    #   A token to continue pagination if more results are available.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_plan_summaries
+    #   A list of summary information for the training plans.
+    #   @return [Array<Types::TrainingPlanSummary>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ListTrainingPlansResponse AWS API Documentation
+    #
+    class ListTrainingPlansResponse < Struct.new(
+      :next_token,
+      :training_plan_summaries)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] creation_time_after
     #   A filter that returns only transform jobs created after the
     #   specified time.
@@ -34223,8 +35575,8 @@ module Aws::SageMaker
     #   @return [Integer]
     #
     # @!attribute [rw] volume_kms_key_id
-    #   The Key Management Service (KMS) key that Amazon SageMaker uses to
-    #   encrypt data on the storage volume attached to the ML compute
+    #   The Key Management Service (KMS) key that Amazon SageMaker AI uses
+    #   to encrypt data on the storage volume attached to the ML compute
     #   instance(s) that run the model monitoring job.
     #   @return [String]
     #
@@ -34393,7 +35745,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] monitoring_inputs
     #   The array of inputs for the monitoring job. Currently we support
-    #   monitoring an Amazon SageMaker Endpoint.
+    #   monitoring an Amazon SageMaker AI Endpoint.
     #   @return [Array<Types::MonitoringInput>]
     #
     # @!attribute [rw] monitoring_output_config
@@ -34427,7 +35779,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] role_arn
     #   The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker
-    #   can assume to perform tasks on your behalf.
+    #   AI can assume to perform tasks on your behalf.
     #   @return [String]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/MonitoringJobDefinition AWS API Documentation
@@ -34548,8 +35900,8 @@ module Aws::SageMaker
     #   @return [Array<Types::MonitoringOutput>]
     #
     # @!attribute [rw] kms_key_id
-    #   The Key Management Service (KMS) key that Amazon SageMaker uses to
-    #   encrypt the model artifacts at rest using Amazon S3 server-side
+    #   The Key Management Service (KMS) key that Amazon SageMaker AI uses
+    #   to encrypt the model artifacts at rest using Amazon S3 server-side
     #   encryption.
     #   @return [String]
     #
@@ -34591,12 +35943,12 @@ module Aws::SageMaker
     #
     # @!attribute [rw] s3_uri
     #   A URI that identifies the Amazon S3 storage location where Amazon
-    #   SageMaker saves the results of a monitoring job.
+    #   SageMaker AI saves the results of a monitoring job.
     #   @return [String]
     #
     # @!attribute [rw] local_path
     #   The local path to the Amazon S3 storage location where Amazon
-    #   SageMaker saves the results of a monitoring job. LocalPath is an
+    #   SageMaker AI saves the results of a monitoring job. LocalPath is an
     #   absolute path for the output data.
     #   @return [String]
     #
@@ -34837,7 +36189,7 @@ module Aws::SageMaker
     # The [VpcConfig][1] configuration object that specifies the VPC that
     # you want the compilation jobs to connect to. For more information on
     # controlling access to your Amazon S3 buckets used for compilation job,
-    # see [Give Amazon SageMaker Compilation Jobs Access to Resources in
+    # see [Give Amazon SageMaker AI Compilation Jobs Access to Resources in
     # Your Amazon VPC][2].
     #
     #
@@ -35010,7 +36362,7 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # Provides summary information for an SageMaker notebook instance.
+    # Provides summary information for an SageMaker AI notebook instance.
     #
     # @!attribute [rw] notebook_instance_name
     #   The name of the notebook instance that you want a summary for.
@@ -35061,7 +36413,8 @@ module Aws::SageMaker
     #   repository in [Amazon Web Services CodeCommit][1] or in any other
     #   Git repository. When you open a notebook instance, it opens in the
     #   directory that contains this repository. For more information, see
-    #   [Associating Git Repositories with SageMaker Notebook Instances][2].
+    #   [Associating Git Repositories with SageMaker AI Notebook
+    #   Instances][2].
     #
     #
     #
@@ -35076,7 +36429,7 @@ module Aws::SageMaker
     #   in [Amazon Web Services CodeCommit][1] or in any other Git
     #   repository. These repositories are cloned at the same level as the
     #   default repository of your notebook instance. For more information,
-    #   see [Associating Git Repositories with SageMaker Notebook
+    #   see [Associating Git Repositories with SageMaker AI Notebook
     #   Instances][2].
     #
     #
@@ -35725,7 +37078,7 @@ module Aws::SageMaker
     # recommended to use for particular TargetPlatform.
     #
     # @!attribute [rw] s3_output_location
-    #   Identifies the S3 bucket where you want Amazon SageMaker to store
+    #   Identifies the S3 bucket where you want Amazon SageMaker AI to store
     #   the model artifacts. For example,
     #   `s3://bucket-name/key-name-prefix`.
     #   @return [String]
@@ -35870,9 +37223,9 @@ module Aws::SageMaker
     #
     # @!attribute [rw] kms_key_id
     #   The Amazon Web Services Key Management Service key (Amazon Web
-    #   Services KMS) that Amazon SageMaker uses to encrypt your output
+    #   Services KMS) that Amazon SageMaker AI uses to encrypt your output
     #   models with Amazon S3 server-side encryption after compilation job.
-    #   If you don't provide a KMS key ID, Amazon SageMaker uses the
+    #   If you don't provide a KMS key ID, Amazon SageMaker AI uses the
     #   default KMS key for Amazon S3 for your role's account. For more
     #   information, see [KMS-Managed Encryption Keys][1] in the *Amazon
     #   Simple Storage Service Developer Guide.*
@@ -36183,6 +37536,82 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Configuration settings for the SageMaker Partner AI App.
+    #
+    # @!attribute [rw] admin_users
+    #   The list of users that are given admin access to the SageMaker
+    #   Partner AI App.
+    #   @return [Array<String>]
+    #
+    # @!attribute [rw] arguments
+    #   This is a map of required inputs for a SageMaker Partner AI App.
+    #   Based on the application type, the map is populated with a key and
+    #   value pair that is specific to the user and application.
+    #   @return [Hash<String,String>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/PartnerAppConfig AWS API Documentation
+    #
+    class PartnerAppConfig < Struct.new(
+      :admin_users,
+      :arguments)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Maintenance configuration settings for the SageMaker Partner AI App.
+    #
+    # @!attribute [rw] maintenance_window_start
+    #   The day and time of the week in Coordinated Universal Time (UTC)
+    #   24-hour standard time that weekly maintenance updates are scheduled.
+    #   This value must take the following format:
+    #   `3-letter-day:24-h-hour:minute`. For example: `TUE:03:30`.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/PartnerAppMaintenanceConfig AWS API Documentation
+    #
+    class PartnerAppMaintenanceConfig < Struct.new(
+      :maintenance_window_start)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A subset of information related to a SageMaker Partner AI App. This
+    # information is used as part of the `ListPartnerApps` API response.
+    #
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] name
+    #   The name of the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] type
+    #   The type of SageMaker Partner AI App to create. Must be one of the
+    #   following: `lakera-guard`, `comet`, `deepchecks-llm-evaluation`, or
+    #   `fiddler`.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The status of the SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] creation_time
+    #   The creation time of the SageMaker Partner AI App.
+    #   @return [Time]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/PartnerAppSummary AWS API Documentation
+    #
+    class PartnerAppSummary < Struct.new(
+      :arn,
+      :name,
+      :type,
+      :status,
+      :creation_time)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The summary of an in-progress deployment when an endpoint is creating
     # or updating with a new endpoint configuration.
     #
@@ -36881,6 +38310,30 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Priority class configuration. When included in `PriorityClasses`,
+    # these class configurations define how tasks are queued.
+    #
+    # @!attribute [rw] name
+    #   Name of the priority class.
+    #   @return [String]
+    #
+    # @!attribute [rw] weight
+    #   Weight of the priority class. The value is within a range from 0 to
+    #   100, where 0 is the default.
+    #
+    #   A weight of 0 is the lowest priority and 100 is the highest. Weight
+    #   0 is the default.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/PriorityClass AWS API Documentation
+    #
+    class PriorityClass < Struct.new(
+      :name,
+      :weight)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Configuration for the cluster used to run a processing job.
     #
     # @!attribute [rw] instance_count
@@ -38668,12 +40121,12 @@ module Aws::SageMaker
     # A collection of settings that apply to an `RSessionGateway` app.
     #
     # @!attribute [rw] default_resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
     # @!attribute [rw] custom_images
-    #   A list of custom SageMaker images that are configured to run as a
+    #   A list of custom SageMaker AI images that are configured to run as a
     #   RSession app.
     #   @return [Array<Types::CustomImage>]
     #
@@ -38727,7 +40180,7 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] default_resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
@@ -38750,7 +40203,7 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] default_resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
@@ -39542,6 +40995,117 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Details about a reserved capacity offering for a training plan
+    # offering.
+    #
+    # For more information about how to reserve GPU capacity for your
+    # SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see
+    # ` CreateTrainingPlan `.
+    #
+    # @!attribute [rw] instance_type
+    #   The instance type for the reserved capacity offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_count
+    #   The number of instances in the reserved capacity offering.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] availability_zone
+    #   The availability zone for the reserved capacity offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] duration_hours
+    #   The number of whole hours in the total duration for this reserved
+    #   capacity offering.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] duration_minutes
+    #   The additional minutes beyond whole hours in the total duration for
+    #   this reserved capacity offering.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] start_time
+    #   The start time of the reserved capacity offering.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time
+    #   The end time of the reserved capacity offering.
+    #   @return [Time]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ReservedCapacityOffering AWS API Documentation
+    #
+    class ReservedCapacityOffering < Struct.new(
+      :instance_type,
+      :instance_count,
+      :availability_zone,
+      :duration_hours,
+      :duration_minutes,
+      :start_time,
+      :end_time)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Details of a reserved capacity for the training plan.
+    #
+    # For more information about how to reserve GPU capacity for your
+    # SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see
+    # ` CreateTrainingPlan `.
+    #
+    # @!attribute [rw] reserved_capacity_arn
+    #   The Amazon Resource Name (ARN); of the reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_type
+    #   The instance type for the reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] total_instance_count
+    #   The total number of instances in the reserved capacity.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] status
+    #   The current status of the reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] availability_zone
+    #   The availability zone for the reserved capacity.
+    #   @return [String]
+    #
+    # @!attribute [rw] duration_hours
+    #   The number of whole hours in the total duration for this reserved
+    #   capacity.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] duration_minutes
+    #   The additional minutes beyond whole hours in the total duration for
+    #   this reserved capacity.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] start_time
+    #   The start time of the reserved capacity.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time
+    #   The end time of the reserved capacity.
+    #   @return [Time]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ReservedCapacitySummary AWS API Documentation
+    #
+    class ReservedCapacitySummary < Struct.new(
+      :reserved_capacity_arn,
+      :instance_type,
+      :total_instance_count,
+      :status,
+      :availability_zone,
+      :duration_hours,
+      :duration_minutes,
+      :start_time,
+      :end_time)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # The resolved attributes.
     #
     # @!attribute [rw] auto_ml_job_objective
@@ -39723,6 +41287,11 @@ module Aws::SageMaker
     #   The configuration of a heterogeneous cluster in JSON format.
     #   @return [Array<Types::InstanceGroup>]
     #
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan to use for this
+    #   resource configuration.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ResourceConfig AWS API Documentation
     #
     class ResourceConfig < Struct.new(
@@ -39731,7 +41300,8 @@ module Aws::SageMaker
       :volume_size_in_gb,
       :volume_kms_key_id,
       :keep_alive_period_in_seconds,
-      :instance_groups)
+      :instance_groups,
+      :training_plan_arn)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -39820,11 +41390,44 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # Specifies the ARN's of a SageMaker image and SageMaker image version,
-    # and the instance type that the version runs on.
+    # Resource sharing configuration.
+    #
+    # @!attribute [rw] strategy
+    #   The strategy of how idle compute is shared within the cluster. The
+    #   following are the options of strategies.
+    #
+    #   * `DontLend`: entities do not lend idle compute.
+    #
+    #   * `Lend`: entities can lend idle compute to entities that can
+    #     borrow.
+    #
+    #   * `LendandBorrow`: entities can lend idle compute and borrow idle
+    #     compute from other entities.
+    #
+    #   Default is `LendandBorrow`.
+    #   @return [String]
+    #
+    # @!attribute [rw] borrow_limit
+    #   The limit on how much idle compute can be borrowed.The values can be
+    #   1 - 500 percent of idle compute that the team is allowed to borrow.
+    #
+    #   Default is `50`.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/ResourceSharingConfig AWS API Documentation
+    #
+    class ResourceSharingConfig < Struct.new(
+      :strategy,
+      :borrow_limit)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Specifies the ARN's of a SageMaker AI image and SageMaker AI image
+    # version, and the instance type that the version runs on.
     #
     # @!attribute [rw] sage_maker_image_arn
-    #   The ARN of the SageMaker image that the image version belongs to.
+    #   The ARN of the SageMaker AI image that the image version belongs to.
     #   @return [String]
     #
     # @!attribute [rw] sage_maker_image_version_arn
@@ -40388,7 +41991,7 @@ module Aws::SageMaker
     #     requested time to run the execution.
     #
     #   * We recommend that if you would like a daily schedule, you do not
-    #     provide this parameter. Amazon SageMaker will pick a time for
+    #     provide this parameter. Amazon SageMaker AI will pick a time for
     #     running every day.
     #
     #    </note>
@@ -40440,6 +42043,35 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # Cluster policy configuration. This policy is used for task
+    # prioritization and fair-share allocation. This helps prioritize
+    # critical workloads and distributes idle compute across entities.
+    #
+    # @!attribute [rw] priority_classes
+    #   List of the priority classes, `PriorityClass`, of the cluster
+    #   policy. When specified, these class configurations define how tasks
+    #   are queued.
+    #   @return [Array<Types::PriorityClass>]
+    #
+    # @!attribute [rw] fair_share
+    #   When enabled, entities borrow idle compute based on their assigned
+    #   `FairShareWeight`.
+    #
+    #   When disabled, entities borrow idle compute based on a first-come
+    #   first-serve basis.
+    #
+    #   Default is `Enabled`.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SchedulerConfig AWS API Documentation
+    #
+    class SchedulerConfig < Struct.new(
+      :priority_classes,
+      :fair_share)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # A multi-expression that searches for the specified resource or
     # resources in a search. All resource objects that satisfy the
     # expression's condition are included in the search results. You must
@@ -40673,6 +42305,76 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] instance_type
+    #   The type of instance you want to search for in the available
+    #   training plan offerings. This field allows you to filter the search
+    #   results based on the specific compute resources you require for your
+    #   SageMaker training jobs or SageMaker HyperPod clusters. When
+    #   searching for training plan offerings, specifying the instance type
+    #   helps you find Reserved Instances that match your computational
+    #   needs.
+    #   @return [String]
+    #
+    # @!attribute [rw] instance_count
+    #   The number of instances you want to reserve in the training plan
+    #   offerings. This allows you to specify the quantity of compute
+    #   resources needed for your SageMaker training jobs or SageMaker
+    #   HyperPod clusters, helping you find reserved capacity offerings that
+    #   match your requirements.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] start_time_after
+    #   A filter to search for training plan offerings with a start time
+    #   after a specified date.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time_before
+    #   A filter to search for reserved capacity offerings with an end time
+    #   before a specified date.
+    #   @return [Time]
+    #
+    # @!attribute [rw] duration_hours
+    #   The desired duration in hours for the training plan offerings.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] target_resources
+    #   The target resources (e.g., SageMaker Training Jobs, SageMaker
+    #   HyperPod) to search for in the offerings.
+    #
+    #   Training plans are specific to their target resource.
+    #
+    #   * A training plan designed for SageMaker training jobs can only be
+    #     used to schedule and run training jobs.
+    #
+    #   * A training plan for HyperPod clusters can be used exclusively to
+    #     provide compute resources to a cluster's instance group.
+    #   @return [Array<String>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SearchTrainingPlanOfferingsRequest AWS API Documentation
+    #
+    class SearchTrainingPlanOfferingsRequest < Struct.new(
+      :instance_type,
+      :instance_count,
+      :start_time_after,
+      :end_time_before,
+      :duration_hours,
+      :target_resources)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] training_plan_offerings
+    #   A list of training plan offerings that match the search criteria.
+    #   @return [Array<Types::TrainingPlanOffering>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SearchTrainingPlanOfferingsResponse AWS API Documentation
+    #
+    class SearchTrainingPlanOfferingsResponse < Struct.new(
+      :training_plan_offerings)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # An array element of `SecondaryStatusTransitions` for
     # [DescribeTrainingJob][1]. It provides additional details about a
     # status that the training job has transitioned through. A training job
@@ -41121,8 +42823,8 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # Specifies options for sharing Amazon SageMaker Studio notebooks. These
-    # settings are specified as part of `DefaultUserSettings` when the
+    # Specifies options for sharing Amazon SageMaker AI Studio notebooks.
+    # These settings are specified as part of `DefaultUserSettings` when the
     # `CreateDomain` API is called, and as part of `UserSettings` when the
     # `CreateUserProfile` API is called. When `SharingSettings` is not
     # specified, notebook sharing isn't allowed.
@@ -41285,7 +42987,7 @@ module Aws::SageMaker
     # The application settings for a Code Editor space.
     #
     # @!attribute [rw] default_resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
@@ -41375,7 +43077,7 @@ module Aws::SageMaker
     # The settings for the JupyterLab application within a space.
     #
     # @!attribute [rw] default_resource_spec
-    #   Specifies the ARN's of a SageMaker image and SageMaker image
+    #   Specifies the ARN's of a SageMaker AI image and SageMaker AI image
     #   version, and the instance type that the version runs on.
     #   @return [Types::ResourceSpec]
     #
@@ -41419,6 +43121,13 @@ module Aws::SageMaker
     #
     # @!attribute [rw] app_type
     #   The type of app created within the space.
+    #
+    #   If using the [ UpdateSpace][1] API, you can't change the app type
+    #   of your space by specifying a different value for this field.
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_UpdateSpace.html
     #   @return [String]
     #
     # @!attribute [rw] space_storage_settings
@@ -41427,8 +43136,8 @@ module Aws::SageMaker
     #
     # @!attribute [rw] custom_file_systems
     #   A file system, created by you, that you assign to a space for an
-    #   Amazon SageMaker Domain. Permitted users can access this file system
-    #   in Amazon SageMaker Studio.
+    #   Amazon SageMaker AI Domain. Permitted users can access this file
+    #   system in Amazon SageMaker AI Studio.
     #   @return [Array<Types::CustomFileSystem>]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/SpaceSettings AWS API Documentation
@@ -42029,23 +43738,23 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
-    # Details of the Amazon SageMaker Studio Lifecycle Configuration.
+    # Details of the Amazon SageMaker AI Studio Lifecycle Configuration.
     #
     # @!attribute [rw] studio_lifecycle_config_arn
     #   The Amazon Resource Name (ARN) of the Lifecycle Configuration.
     #   @return [String]
     #
     # @!attribute [rw] studio_lifecycle_config_name
-    #   The name of the Amazon SageMaker Studio Lifecycle Configuration.
+    #   The name of the Amazon SageMaker AI Studio Lifecycle Configuration.
     #   @return [String]
     #
     # @!attribute [rw] creation_time
-    #   The creation time of the Amazon SageMaker Studio Lifecycle
+    #   The creation time of the Amazon SageMaker AI Studio Lifecycle
     #   Configuration.
     #   @return [Time]
     #
     # @!attribute [rw] last_modified_time
-    #   This value is equivalent to CreationTime because Amazon SageMaker
+    #   This value is equivalent to CreationTime because Amazon SageMaker AI
     #   Studio Lifecycle Configurations are immutable.
     #   @return [Time]
     #
@@ -42440,7 +44149,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] default_resource_spec
     #   The default instance type and the Amazon Resource Name (ARN) of the
-    #   SageMaker image created on the instance.
+    #   SageMaker AI image created on the instance.
     #   @return [Types::ResourceSpec]
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TensorBoardAppSettings AWS API Documentation
@@ -43651,6 +45360,15 @@ module Aws::SageMaker
     #   The status of the warm pool associated with the training job.
     #   @return [Types::WarmPoolStatus]
     #
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan associated with
+    #   this training job.
+    #
+    #   For more information about how to reserve GPU capacity for your
+    #   SageMaker HyperPod clusters using Amazon SageMaker Training Plan,
+    #   see ` CreateTrainingPlan `.
+    #   @return [String]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingJobSummary AWS API Documentation
     #
     class TrainingJobSummary < Struct.new(
@@ -43661,7 +45379,209 @@ module Aws::SageMaker
       :last_modified_time,
       :training_job_status,
       :secondary_status,
-      :warm_pool_status)
+      :warm_pool_status,
+      :training_plan_arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A filter to apply when listing or searching for training plans.
+    #
+    # For more information about how to reserve GPU capacity for your
+    # SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see
+    # ` CreateTrainingPlan `.
+    #
+    # @!attribute [rw] name
+    #   The name of the filter field (e.g., Status, InstanceType).
+    #   @return [String]
+    #
+    # @!attribute [rw] value
+    #   The value to filter by for the specified field.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingPlanFilter AWS API Documentation
+    #
+    class TrainingPlanFilter < Struct.new(
+      :name,
+      :value)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Details about a training plan offering.
+    #
+    # For more information about how to reserve GPU capacity for your
+    # SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see
+    # ` CreateTrainingPlan `.
+    #
+    # @!attribute [rw] training_plan_offering_id
+    #   The unique identifier for this training plan offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] target_resources
+    #   The target resources (e.g., SageMaker Training Jobs, SageMaker
+    #   HyperPod) for this training plan offering.
+    #
+    #   Training plans are specific to their target resource.
+    #
+    #   * A training plan designed for SageMaker training jobs can only be
+    #     used to schedule and run training jobs.
+    #
+    #   * A training plan for HyperPod clusters can be used exclusively to
+    #     provide compute resources to a cluster's instance group.
+    #   @return [Array<String>]
+    #
+    # @!attribute [rw] requested_start_time_after
+    #   The requested start time that the user specified when searching for
+    #   the training plan offering.
+    #   @return [Time]
+    #
+    # @!attribute [rw] requested_end_time_before
+    #   The requested end time that the user specified when searching for
+    #   the training plan offering.
+    #   @return [Time]
+    #
+    # @!attribute [rw] duration_hours
+    #   The number of whole hours in the total duration for this training
+    #   plan offering.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] duration_minutes
+    #   The additional minutes beyond whole hours in the total duration for
+    #   this training plan offering.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] upfront_fee
+    #   The upfront fee for this training plan offering.
+    #   @return [String]
+    #
+    # @!attribute [rw] currency_code
+    #   The currency code for the upfront fee (e.g., USD).
+    #   @return [String]
+    #
+    # @!attribute [rw] reserved_capacity_offerings
+    #   A list of reserved capacity offerings associated with this training
+    #   plan offering.
+    #   @return [Array<Types::ReservedCapacityOffering>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingPlanOffering AWS API Documentation
+    #
+    class TrainingPlanOffering < Struct.new(
+      :training_plan_offering_id,
+      :target_resources,
+      :requested_start_time_after,
+      :requested_end_time_before,
+      :duration_hours,
+      :duration_minutes,
+      :upfront_fee,
+      :currency_code,
+      :reserved_capacity_offerings)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Details of the training plan.
+    #
+    # For more information about how to reserve GPU capacity for your
+    # SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see
+    # ` CreateTrainingPlan `.
+    #
+    # @!attribute [rw] training_plan_arn
+    #   The Amazon Resource Name (ARN); of the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] training_plan_name
+    #   The name of the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] status
+    #   The current status of the training plan (e.g., Pending, Active,
+    #   Expired). To see the complete list of status values available for a
+    #   training plan, refer to the `Status` attribute within the `
+    #   TrainingPlanSummary ` object.
+    #   @return [String]
+    #
+    # @!attribute [rw] status_message
+    #   A message providing additional information about the current status
+    #   of the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] duration_hours
+    #   The number of whole hours in the total duration for this training
+    #   plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] duration_minutes
+    #   The additional minutes beyond whole hours in the total duration for
+    #   this training plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] start_time
+    #   The start time of the training plan.
+    #   @return [Time]
+    #
+    # @!attribute [rw] end_time
+    #   The end time of the training plan.
+    #   @return [Time]
+    #
+    # @!attribute [rw] upfront_fee
+    #   The upfront fee for the training plan.
+    #   @return [String]
+    #
+    # @!attribute [rw] currency_code
+    #   The currency code for the upfront fee (e.g., USD).
+    #   @return [String]
+    #
+    # @!attribute [rw] total_instance_count
+    #   The total number of instances reserved in this training plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] available_instance_count
+    #   The number of instances currently available for use in this training
+    #   plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] in_use_instance_count
+    #   The number of instances currently in use from this training plan.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] target_resources
+    #   The target resources (e.g., training jobs, HyperPod clusters) that
+    #   can use this training plan.
+    #
+    #   Training plans are specific to their target resource.
+    #
+    #   * A training plan designed for SageMaker training jobs can only be
+    #     used to schedule and run training jobs.
+    #
+    #   * A training plan for HyperPod clusters can be used exclusively to
+    #     provide compute resources to a cluster's instance group.
+    #   @return [Array<String>]
+    #
+    # @!attribute [rw] reserved_capacity_summaries
+    #   A list of reserved capacities associated with this training plan,
+    #   including details such as instance types, counts, and availability
+    #   zones.
+    #   @return [Array<Types::ReservedCapacitySummary>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/TrainingPlanSummary AWS API Documentation
+    #
+    class TrainingPlanSummary < Struct.new(
+      :training_plan_arn,
+      :training_plan_name,
+      :status,
+      :status_message,
+      :duration_hours,
+      :duration_minutes,
+      :start_time,
+      :end_time,
+      :upfront_fee,
+      :currency_code,
+      :total_instance_count,
+      :available_instance_count,
+      :in_use_instance_count,
+      :target_resources,
+      :reserved_capacity_summaries)
       SENSITIVE = []
       include Aws::Structure
     end
@@ -45316,6 +47236,50 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] cluster_scheduler_config_id
+    #   ID of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] target_version
+    #   Target version.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] scheduler_config
+    #   Cluster policy configuration.
+    #   @return [Types::SchedulerConfig]
+    #
+    # @!attribute [rw] description
+    #   Description of the cluster policy.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdateClusterSchedulerConfigRequest AWS API Documentation
+    #
+    class UpdateClusterSchedulerConfigRequest < Struct.new(
+      :cluster_scheduler_config_id,
+      :target_version,
+      :scheduler_config,
+      :description)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] cluster_scheduler_config_arn
+    #   ARN of the cluster policy.
+    #   @return [String]
+    #
+    # @!attribute [rw] cluster_scheduler_config_version
+    #   Version of the cluster policy.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdateClusterSchedulerConfigResponse AWS API Documentation
+    #
+    class UpdateClusterSchedulerConfigResponse < Struct.new(
+      :cluster_scheduler_config_arn,
+      :cluster_scheduler_config_version)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] cluster_name
     #   Specify the name or the Amazon Resource Name (ARN) of the SageMaker
     #   HyperPod cluster you want to update for security patching.
@@ -45377,6 +47341,65 @@ module Aws::SageMaker
       include Aws::Structure
     end
 
+    # @!attribute [rw] compute_quota_id
+    #   ID of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] target_version
+    #   Target version.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] compute_quota_config
+    #   Configuration of the compute allocation definition. This includes
+    #   the resource sharing option, and the setting to preempt low priority
+    #   tasks.
+    #   @return [Types::ComputeQuotaConfig]
+    #
+    # @!attribute [rw] compute_quota_target
+    #   The target entity to allocate compute resources to.
+    #   @return [Types::ComputeQuotaTarget]
+    #
+    # @!attribute [rw] activation_state
+    #   The state of the compute allocation being described. Use to enable
+    #   or disable compute allocation.
+    #
+    #   Default is `Enabled`.
+    #   @return [String]
+    #
+    # @!attribute [rw] description
+    #   Description of the compute allocation definition.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdateComputeQuotaRequest AWS API Documentation
+    #
+    class UpdateComputeQuotaRequest < Struct.new(
+      :compute_quota_id,
+      :target_version,
+      :compute_quota_config,
+      :compute_quota_target,
+      :activation_state,
+      :description)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] compute_quota_arn
+    #   ARN of the compute allocation definition.
+    #   @return [String]
+    #
+    # @!attribute [rw] compute_quota_version
+    #   Version of the compute allocation definition.
+    #   @return [Integer]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdateComputeQuotaResponse AWS API Documentation
+    #
+    class UpdateComputeQuotaResponse < Struct.new(
+      :compute_quota_arn,
+      :compute_quota_version)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] context_name
     #   The name of the context to update.
     #   @return [String]
@@ -45507,7 +47530,7 @@ module Aws::SageMaker
     #   Specifies the VPC used for non-EFS traffic.
     #
     #   * `PublicInternetOnly` - Non-EFS traffic is through a VPC managed by
-    #     Amazon SageMaker, which allows direct internet access.
+    #     Amazon SageMaker AI, which allows direct internet access.
     #
     #   * `VpcOnly` - All Studio traffic is through the specified VPC and
     #     subnets.
@@ -45829,7 +47852,7 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] role_arn
-    #   The new ARN for the IAM role that enables Amazon SageMaker to
+    #   The new ARN for the IAM role that enables Amazon SageMaker AI to
     #   perform tasks on your behalf.
     #   @return [String]
     #
@@ -45894,16 +47917,16 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] job_type
-    #   Indicates SageMaker job type compatibility.
+    #   Indicates SageMaker AI job type compatibility.
     #
-    #   * `TRAINING`: The image version is compatible with SageMaker
+    #   * `TRAINING`: The image version is compatible with SageMaker AI
     #     training jobs.
     #
-    #   * `INFERENCE`: The image version is compatible with SageMaker
+    #   * `INFERENCE`: The image version is compatible with SageMaker AI
     #     inference jobs.
     #
     #   * `NOTEBOOK_KERNEL`: The image version is compatible with SageMaker
-    #     notebook kernels.
+    #     AI notebook kernels.
     #   @return [String]
     #
     # @!attribute [rw] ml_framework
@@ -46382,11 +48405,11 @@ module Aws::SageMaker
     #   @return [String]
     #
     # @!attribute [rw] role_arn
-    #   The Amazon Resource Name (ARN) of the IAM role that SageMaker can
+    #   The Amazon Resource Name (ARN) of the IAM role that SageMaker AI can
     #   assume to access the notebook instance. For more information, see
-    #   [SageMaker Roles][1].
+    #   [SageMaker AI Roles][1].
     #
-    #   <note markdown="1"> To be able to pass this role to SageMaker, the caller of this API
+    #   <note markdown="1"> To be able to pass this role to SageMaker AI, the caller of this API
     #   must have the `iam:PassRole` permission.
     #
     #    </note>
@@ -46417,7 +48440,7 @@ module Aws::SageMaker
     # @!attribute [rw] volume_size_in_gb
     #   The size, in GB, of the ML storage volume to attach to the notebook
     #   instance. The default value is 5 GB. ML storage volumes are
-    #   encrypted, so SageMaker can't determine the amount of available
+    #   encrypted, so SageMaker AI can't determine the amount of available
     #   free space on the volume. Because of this, you can increase the
     #   volume size when you update a notebook instance, but you can't
     #   decrease the volume size. If you want to decrease the size of the ML
@@ -46432,7 +48455,8 @@ module Aws::SageMaker
     #   repository in [Amazon Web Services CodeCommit][1] or in any other
     #   Git repository. When you open a notebook instance, it opens in the
     #   directory that contains this repository. For more information, see
-    #   [Associating Git Repositories with SageMaker Notebook Instances][2].
+    #   [Associating Git Repositories with SageMaker AI Notebook
+    #   Instances][2].
     #
     #
     #
@@ -46447,7 +48471,7 @@ module Aws::SageMaker
     #   in [Amazon Web Services CodeCommit][1] or in any other Git
     #   repository. These repositories are cloned at the same level as the
     #   default repository of your notebook instance. For more information,
-    #   see [Associating Git Repositories with SageMaker Notebook
+    #   see [Associating Git Repositories with SageMaker AI Notebook
     #   Instances][2].
     #
     #
@@ -46555,6 +48579,68 @@ module Aws::SageMaker
     #
     class UpdateNotebookInstanceOutput < Aws::EmptyStructure; end
 
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App to update.
+    #   @return [String]
+    #
+    # @!attribute [rw] maintenance_config
+    #   Maintenance configuration settings for the SageMaker Partner AI App.
+    #   @return [Types::PartnerAppMaintenanceConfig]
+    #
+    # @!attribute [rw] tier
+    #   Indicates the instance type and size of the cluster attached to the
+    #   SageMaker Partner AI App.
+    #   @return [String]
+    #
+    # @!attribute [rw] application_config
+    #   Configuration settings for the SageMaker Partner AI App.
+    #   @return [Types::PartnerAppConfig]
+    #
+    # @!attribute [rw] enable_iam_session_based_identity
+    #   When set to `TRUE`, the SageMaker Partner AI App sets the Amazon Web
+    #   Services IAM session name or the authenticated IAM user as the
+    #   identity of the SageMaker Partner AI App user.
+    #   @return [Boolean]
+    #
+    # @!attribute [rw] client_token
+    #   A unique token that guarantees that the call to this API is
+    #   idempotent.
+    #
+    #   **A suitable default value is auto-generated.** You should normally
+    #   not need to pass this option.
+    #   @return [String]
+    #
+    # @!attribute [rw] tags
+    #   Each tag consists of a key and an optional value. Tag keys must be
+    #   unique per resource.
+    #   @return [Array<Types::Tag>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdatePartnerAppRequest AWS API Documentation
+    #
+    class UpdatePartnerAppRequest < Struct.new(
+      :arn,
+      :maintenance_config,
+      :tier,
+      :application_config,
+      :enable_iam_session_based_identity,
+      :client_token,
+      :tags)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] arn
+    #   The ARN of the SageMaker Partner AI App that was updated.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-2017-07-24/UpdatePartnerAppResponse AWS API Documentation
+    #
+    class UpdatePartnerAppResponse < Struct.new(
+      :arn)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # @!attribute [rw] pipeline_execution_arn
     #   The Amazon Resource Name (ARN) of the pipeline execution.
     #   @return [String]
@@ -47186,8 +49272,8 @@ module Aws::SageMaker
     #   set to `VpcOnly`, unless specified as part of the
     #   `DefaultUserSettings` for the domain.
     #
-    #   Amazon SageMaker adds a security group to allow NFS traffic from
-    #   Amazon SageMaker Studio. Therefore, the number of security groups
+    #   Amazon SageMaker AI adds a security group to allow NFS traffic from
+    #   Amazon SageMaker AI Studio. Therefore, the number of security groups
     #   that you can specify is one less than the maximum number shown.
     #
     #   SageMaker applies these settings only to private spaces that the
@@ -47196,7 +49282,7 @@ module Aws::SageMaker
     #   @return [Array<String>]
     #
     # @!attribute [rw] sharing_settings
-    #   Specifies options for sharing Amazon SageMaker Studio notebooks.
+    #   Specifies options for sharing Amazon SageMaker AI Studio notebooks.
     #   @return [Types::SharingSettings]
     #
     # @!attribute [rw] jupyter_server_app_settings
@@ -47279,7 +49365,7 @@ module Aws::SageMaker
     #
     # @!attribute [rw] custom_file_system_configs
     #   The settings for assigning a custom file system to a user profile.
-    #   Permitted users can access this file system in Amazon SageMaker
+    #   Permitted users can access this file system in Amazon SageMaker AI
     #   Studio.
     #
     #   SageMaker applies these settings only to private spaces that the