@aws-sdk/client-sagemaker 3.1054.0 → 3.1056.0

package/dist-types/models/models_1.d.ts

@@ -1,6 +1,105 @@
 import { AutomaticJsonStringConversion as __AutomaticJsonStringConversion } from "@smithy/core/serde";
-import type { _InstanceType, AccountDefaultStatus, ActionStatus, ActivationState, AppInstanceType, AppNetworkAccessType, AppSecurityGroupManagement, AppType, AuthMode, AutoMountHomeEFS, AvailabilityZoneBalanceEnforcementMode, AwsManagedHumanLoopRequestSource, CapacityReservationPreference, ClusterNodeProvisioningMode, ClusterNodeRecovery, CollectionType, CompleteOnConvergence, ContainerMode, ContentClassifier, DeviceSubsetType, DirectInternetAccess, EdgePresetDeploymentType, ExecutionRoleIdentityConfig, ExecutionRoleSessionNameMode, FailureHandlingPolicy, FairShare, FeatureStatus, FeatureType, FlatInvocations, Framework, HomeEfsFileSystemCreation, HubContentType, HyperParameterScalingType, HyperParameterTuningAllocationStrategy, HyperParameterTuningJobObjectiveType, HyperParameterTuningJobStrategyType, HyperParameterTuningJobWarmStartType, IdleResourceSharing, InferenceComponentPlacementStrategy, InferenceExecutionMode, InferenceExperimentType, IPAddressType, JobType, ManagedInstanceScalingScaleInStrategy, ManagedInstanceScalingStatus, ManagedStorageType, MetricPublishFrequencyInSeconds, MlTools, ModelApprovalStatus, ModelCacheSetting, ModelCardStatus, ModelInfrastructureType, ModelPackageRegistrationType, ModelRegistrationMode, ModelSpeculativeDecodingS3DataType, ModelSpeculativeDecodingTechnique, MonitoringProblemType, MonitoringType, NotebookInstanceAcceleratorType, NotebookOutputOption, OptimizationJobDeploymentInstanceType, ParameterType, ProblemType, ProcessingInstanceType, ProcessingS3DataDistributionType, ProcessingS3InputMode, ProcessingS3UploadMode, Processor, ProductionVariantAcceleratorType, ProductionVariantInferenceAmiVersion, ProductionVariantInstanceType, RecommendationJobSupportedEndpointType, RecommendationJobType, RootAccess, RoutingStrategy, RStudioServerProAccessStatus, RStudioServerProUserGroup, SageMakerImageName, SkipModelValidation, StorageType, StudioWebPortal, TableFormat, TagPropagation, TargetDevice, TargetPlatformAccelerator, TargetPlatformArch, TargetPlatformOs, ThroughputMode, TrackingServerSize, TrafficType, TrainingInputMode, TrainingInstanceType, TrainingJobEarlyStoppingType, TtlDurationUnit, VendorGuidance } from "./enums";
-import type { ActionSource, AdditionalInferenceSpecificationDefinition, AdditionalModelDataSource, AdditionalS3DataSource, AIBenchmarkNetworkConfig, AIBenchmarkOutputConfig, AIBenchmarkTarget, AIDatasetConfig, AIModelSource, AIRecommendationComputeSpec, AIRecommendationInferenceSpecification, AIRecommendationOutputConfig, AIRecommendationPerformanceTarget, AIWorkloadConfigs, AlgorithmValidationSpecification, AmazonQSettings, AnnotationConsolidationConfig, AppLifecycleManagement, ArtifactSource, AsyncInferenceConfig, AuthorizedUrl, AutoMLChannel, AutoMLComputeConfig, AutoMLDataSplitConfig, AutoMLJobChannel, AutoMLJobConfig, AutoMLJobObjective, AutoMLOutputDataConfig, AutoMLProblemTypeConfig, AutoMLSecurityConfig, AutoParameter, AutoRollbackConfig, Autotune, BatchTransformInput, BestObjectiveNotImproving, Bias, BlueGreenUpdatePolicy, CanvasAppSettings, CapacitySize, CaptureContentTypeHeader, CaptureOption, CategoricalParameter, CategoricalParameterRange, CategoricalParameterRangeSpecification, Channel, ChannelSpecification, CheckpointConfig, ClarifyExplainerConfig, ClusterAutoScalingConfig, ClusterInstanceGroupSpecification, ClusterOrchestrator, ClusterRestrictedInstanceGroupSpecification, ClusterTieredStorageConfig, CodeEditorAppImageConfig, CodeEditorAppSettings, CodeRepository, CollectionConfig, ComputeQuotaConfig, ComputeQuotaTarget, CustomImage, GitConfig, ImageConfig, InferenceSpecification, JupyterLabAppImageConfig, KernelGatewayImageConfig, MetricDefinition, MetricsSource, ModelDataSource, OutputDataConfig, ResourceConfig, ResourceSpec, StoppingCondition, Tag, TransformJobDefinition, VpcConfig } from "./models_0";
+import type { _InstanceType, AccountDefaultStatus, ActionStatus, ActivationState, AppInstanceType, AppNetworkAccessType, AppSecurityGroupManagement, AppType, AuthMode, AutoMountHomeEFS, AvailabilityZoneBalanceEnforcementMode, AwsManagedHumanLoopRequestSource, CapacityReservationPreference, ClusterNodeProvisioningMode, ClusterNodeRecovery, CollectionType, CompleteOnConvergence, ConditionOutcome, ContainerMode, ContentClassifier, DeviceSubsetType, DirectInternetAccess, EdgePresetDeploymentType, ExecutionRoleIdentityConfig, ExecutionRoleSessionNameMode, FailureHandlingPolicy, FairShare, FeatureStatus, FeatureType, FlatInvocations, Framework, HomeEfsFileSystemCreation, HubContentType, HyperParameterScalingType, HyperParameterTuningAllocationStrategy, HyperParameterTuningJobObjectiveType, HyperParameterTuningJobStrategyType, HyperParameterTuningJobWarmStartType, IdleResourceSharing, InferenceComponentPlacementStrategy, InferenceExecutionMode, InferenceExperimentType, IPAddressType, JobType, ManagedInstanceScalingScaleInStrategy, ManagedInstanceScalingStatus, ManagedStorageType, MetricPublishFrequencyInSeconds, MlTools, ModelApprovalStatus, ModelCacheSetting, ModelCardStatus, ModelInfrastructureType, ModelPackageRegistrationType, ModelRegistrationMode, ModelSpeculativeDecodingS3DataType, ModelSpeculativeDecodingTechnique, MonitoringProblemType, MonitoringType, NotebookInstanceAcceleratorType, NotebookOutputOption, ParameterType, ProblemType, ProcessingInstanceType, ProcessingS3DataDistributionType, ProcessingS3InputMode, ProcessingS3UploadMode, Processor, ProductionVariantAcceleratorType, ProductionVariantInferenceAmiVersion, ProductionVariantInstanceType, RecommendationJobSupportedEndpointType, RecommendationJobType, RepositoryAccessMode, RootAccess, RoutingStrategy, RStudioServerProAccessStatus, RStudioServerProUserGroup, SageMakerImageName, SchedulerResourceStatus, SkipModelValidation, StorageType, StudioWebPortal, TableFormat, TagPropagation, TargetDevice, TargetPlatformAccelerator, TargetPlatformArch, TargetPlatformOs, ThroughputMode, TrackingServerSize, TrafficType, TrainingInputMode, TrainingInstanceType, TrainingJobEarlyStoppingType, TtlDurationUnit, VendorGuidance } from "./enums";
+import type { ActionSource, AdditionalInferenceSpecificationDefinition, AdditionalModelDataSource, AdditionalS3DataSource, AIBenchmarkNetworkConfig, AIBenchmarkOutputConfig, AIBenchmarkTarget, AIDatasetConfig, AIModelSource, AIRecommendationComputeSpec, AIRecommendationInferenceSpecification, AIRecommendationOutputConfig, AIRecommendationPerformanceTarget, AIWorkloadConfigs, AlgorithmValidationSpecification, AmazonQSettings, AnnotationConsolidationConfig, AppLifecycleManagement, ArtifactSource, AsyncInferenceConfig, AuthorizedUrl, AutoMLChannel, AutoMLComputeConfig, AutoMLDataSplitConfig, AutoMLJobChannel, AutoMLJobConfig, AutoMLJobObjective, AutoMLOutputDataConfig, AutoMLProblemTypeConfig, AutoMLSecurityConfig, AutoParameter, AutoRollbackConfig, Autotune, BatchTransformInput, BestObjectiveNotImproving, Bias, BlueGreenUpdatePolicy, CanvasAppSettings, CapacitySize, CaptureContentTypeHeader, CaptureOption, CategoricalParameter, CategoricalParameterRange, CategoricalParameterRangeSpecification, Channel, ChannelSpecification, CheckpointConfig, ClarifyExplainerConfig, ClusterAutoScalingConfig, ClusterInstanceGroupSpecification, ClusterOrchestrator, ClusterRestrictedInstanceGroupsConfig, ClusterRestrictedInstanceGroupSpecification, ClusterTieredStorageConfig, CodeEditorAppImageConfig, CodeEditorAppSettings, CodeRepository, CollectionConfig, ComputeQuotaConfig, ComputeQuotaTarget, CustomImage, GitConfig, InferenceSpecification, JupyterLabAppImageConfig, KernelGatewayImageConfig, MetricDefinition, MetricsSource, ModelDataSource, OutputDataConfig, ResourceConfig, ResourceSpec, StoppingCondition, Tag, TransformJobDefinition, VpcConfig } from "./models_0";
+/**
+ * <p>Summary of the compute allocation definition.</p>
+ * @public
+ */
+export interface ComputeQuotaSummary {
+    /**
+     * <p>ARN of the compute allocation definition.</p>
+     * @public
+     */
+    ComputeQuotaArn: string | undefined;
+    /**
+     * <p>ID of the compute allocation definition.</p>
+     * @public
+     */
+    ComputeQuotaId: string | undefined;
+    /**
+     * <p>Name of the compute allocation definition.</p>
+     * @public
+     */
+    Name: string | undefined;
+    /**
+     * <p>Version of the compute allocation definition.</p>
+     * @public
+     */
+    ComputeQuotaVersion?: number | undefined;
+    /**
+     * <p>Status of the compute allocation definition.</p>
+     * @public
+     */
+    Status: SchedulerResourceStatus | undefined;
+    /**
+     * <p>ARN of the cluster.</p>
+     * @public
+     */
+    ClusterArn?: string | undefined;
+    /**
+     * <p>Configuration of the compute allocation definition. This includes the resource sharing option, and the setting to preempt low priority tasks.</p>
+     * @public
+     */
+    ComputeQuotaConfig?: ComputeQuotaConfig | undefined;
+    /**
+     * <p>The target entity to allocate compute resources to.</p>
+     * @public
+     */
+    ComputeQuotaTarget: ComputeQuotaTarget | undefined;
+    /**
+     * <p>The state of the compute allocation being described. Use to enable or disable compute allocation.</p> <p>Default is <code>Enabled</code>.</p>
+     * @public
+     */
+    ActivationState?: ActivationState | undefined;
+    /**
+     * <p>Creation time of the compute allocation definition.</p>
+     * @public
+     */
+    CreationTime: Date | undefined;
+    /**
+     * <p>Last modified time of the compute allocation definition.</p>
+     * @public
+     */
+    LastModifiedTime?: Date | undefined;
+}
+/**
+ * <p>Metadata for a Condition step.</p>
+ * @public
+ */
+export interface ConditionStepMetadata {
+    /**
+     * <p>The outcome of the Condition step evaluation.</p>
+     * @public
+     */
+    Outcome?: ConditionOutcome | undefined;
+}
+/**
+ * <p>Specifies an authentication configuration for the private docker registry where your model image is hosted. Specify a value for this property only if you specified <code>Vpc</code> as the value for the <code>RepositoryAccessMode</code> field of the <code>ImageConfig</code> object that you passed to a call to <code>CreateModel</code> and the private Docker registry where the model image is hosted requires authentication.</p>
+ * @public
+ */
+export interface RepositoryAuthConfig {
+    /**
+     * <p>The Amazon Resource Name (ARN) of an Amazon Web Services Lambda function that provides credentials to authenticate to the private Docker registry where your model image is hosted. For information about how to create an Amazon Web Services Lambda function, see <a href="https://docs.aws.amazon.com/lambda/latest/dg/getting-started-create-function.html">Create a Lambda function with the console</a> in the <i>Amazon Web Services Lambda Developer Guide</i>.</p>
+     * @public
+     */
+    RepositoryCredentialsProviderArn: string | undefined;
+}
+/**
+ * <p>Specifies whether the model container is in Amazon ECR or a private Docker registry accessible from your Amazon Virtual Private Cloud (VPC).</p>
+ * @public
+ */
+export interface ImageConfig {
+    /**
+     * <p>Set this to one of the following values:</p> <ul> <li> <p> <code>Platform</code> - The model image is hosted in Amazon ECR.</p> </li> <li> <p> <code>Vpc</code> - The model image is hosted in a private Docker registry in your VPC.</p> </li> </ul>
+     * @public
+     */
+    RepositoryAccessMode: RepositoryAccessMode | undefined;
+    /**
+     * <p>(Optional) Specifies an authentication configuration for the private docker registry where your model image is hosted. Specify a value for this property only if you specified <code>Vpc</code> as the value for the <code>RepositoryAccessMode</code> field, and the private Docker registry where the model image is hosted requires authentication.</p>
+     * @public
+     */
+    RepositoryAuthConfig?: RepositoryAuthConfig | undefined;
+}
 /**
  * <p>Specifies additional configuration for hosting multi-model endpoints.</p>
  * @public
@@ -919,6 +1018,11 @@ export interface CreateClusterRequest {
      * @public
      */
     RestrictedInstanceGroups?: ClusterRestrictedInstanceGroupSpecification[] | undefined;
+    /**
+     * <p>The configuration for the restricted instance groups (RIG) in the SageMaker HyperPod cluster.</p>
+     * @public
+     */
+    RestrictedInstanceGroupsConfig?: ClusterRestrictedInstanceGroupsConfig | undefined;
     /**
      * <p>Specifies the Amazon Virtual Private Cloud (VPC) that is associated with the Amazon SageMaker HyperPod cluster. You can control access to and from your resources by configuring your VPC. For more information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/infrastructure-give-access.html">Give SageMaker access to resources in your Amazon VPC</a>.</p> <note> <p>When your Amazon VPC and subnets support IPv6, network communications differ based on the cluster orchestration platform:</p> <ul> <li> <p>Slurm-orchestrated clusters automatically configure nodes with dual IPv6 and IPv4 addresses, allowing immediate IPv6 network communications.</p> </li> <li> <p>In Amazon EKS-orchestrated clusters, nodes receive dual-stack addressing, but pods can only use IPv6 when the Amazon EKS cluster is explicitly IPv6-enabled. For information about deploying an IPv6 Amazon EKS cluster, see <a href="https://docs.aws.amazon.com/eks/latest/userguide/deploy-ipv6-cluster.html#_deploy_an_ipv6_cluster_with_eksctl">Amazon EKS IPv6 Cluster Deployment</a>.</p> </li> </ul> <p>Additional resources for IPv6 configuration:</p> <ul> <li> <p>For information about adding IPv6 support to your VPC, see to <a href="https://docs.aws.amazon.com/vpc/latest/userguide/vpc-migrate-ipv6.html">IPv6 Support for VPC</a>.</p> </li> <li> <p>For information about creating a new IPv6-compatible VPC, see <a href="https://docs.aws.amazon.com/vpc/latest/userguide/create-vpc.html">Amazon VPC Creation Guide</a>.</p> </li> <li> <p>To configure SageMaker HyperPod with a custom Amazon VPC, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-hyperpod-prerequisites.html#sagemaker-hyperpod-prerequisites-optional-vpc">Custom Amazon VPC Setup for SageMaker HyperPod</a>.</p> </li> </ul> </note>
      * @public
@@ -7063,175 +7167,3 @@ export interface ModelSpeculativeDecodingConfig {
      */
     TrainingDataSource?: ModelSpeculativeDecodingTrainingDataSource | undefined;
 }
-/**
- * <p>Settings for an optimization technique that you apply with a model optimization job.</p>
- * @public
- */
-export type OptimizationConfig = OptimizationConfig.ModelCompilationConfigMember | OptimizationConfig.ModelQuantizationConfigMember | OptimizationConfig.ModelShardingConfigMember | OptimizationConfig.ModelSpeculativeDecodingConfigMember | OptimizationConfig.$UnknownMember;
-/**
- * @public
- */
-export declare namespace OptimizationConfig {
-    /**
-     * <p>Settings for the model quantization technique that's applied by a model optimization job.</p>
-     * @public
-     */
-    interface ModelQuantizationConfigMember {
-        ModelQuantizationConfig: ModelQuantizationConfig;
-        ModelCompilationConfig?: never;
-        ModelShardingConfig?: never;
-        ModelSpeculativeDecodingConfig?: never;
-        $unknown?: never;
-    }
-    /**
-     * <p>Settings for the model compilation technique that's applied by a model optimization job.</p>
-     * @public
-     */
-    interface ModelCompilationConfigMember {
-        ModelQuantizationConfig?: never;
-        ModelCompilationConfig: ModelCompilationConfig;
-        ModelShardingConfig?: never;
-        ModelSpeculativeDecodingConfig?: never;
-        $unknown?: never;
-    }
-    /**
-     * <p>Settings for the model sharding technique that's applied by a model optimization job.</p>
-     * @public
-     */
-    interface ModelShardingConfigMember {
-        ModelQuantizationConfig?: never;
-        ModelCompilationConfig?: never;
-        ModelShardingConfig: ModelShardingConfig;
-        ModelSpeculativeDecodingConfig?: never;
-        $unknown?: never;
-    }
-    /**
-     * <p>Settings for the model speculative decoding technique that's applied by a model optimization job.</p>
-     * @public
-     */
-    interface ModelSpeculativeDecodingConfigMember {
-        ModelQuantizationConfig?: never;
-        ModelCompilationConfig?: never;
-        ModelShardingConfig?: never;
-        ModelSpeculativeDecodingConfig: ModelSpeculativeDecodingConfig;
-        $unknown?: never;
-    }
-    /**
-     * @public
-     */
-    interface $UnknownMember {
-        ModelQuantizationConfig?: never;
-        ModelCompilationConfig?: never;
-        ModelShardingConfig?: never;
-        ModelSpeculativeDecodingConfig?: never;
-        $unknown: [string, any];
-    }
-    /**
-     * @deprecated unused in schema-serde mode.
-     *
-     */
-    interface Visitor<T> {
-        ModelQuantizationConfig: (value: ModelQuantizationConfig) => T;
-        ModelCompilationConfig: (value: ModelCompilationConfig) => T;
-        ModelShardingConfig: (value: ModelShardingConfig) => T;
-        ModelSpeculativeDecodingConfig: (value: ModelSpeculativeDecodingConfig) => T;
-        _: (name: string, value: any) => T;
-    }
-}
-/**
- * <p>Details for where to store the optimized model that you create with the optimization job.</p>
- * @public
- */
-export interface OptimizationJobOutputConfig {
-    /**
-     * <p>The Amazon Resource Name (ARN) of a key in Amazon Web Services KMS. SageMaker uses they key to encrypt the artifacts of the optimized model when SageMaker uploads the model to Amazon S3.</p>
-     * @public
-     */
-    KmsKeyId?: string | undefined;
-    /**
-     * <p>The Amazon S3 URI for where to store the optimized model that you create with an optimization job.</p>
-     * @public
-     */
-    S3OutputLocation: string | undefined;
-    /**
-     * <p>The name of a SageMaker model to use as the output destination for an optimization job.</p>
-     * @public
-     */
-    SageMakerModel?: OptimizationSageMakerModel | undefined;
-}
-/**
- * <p>A VPC in Amazon VPC that's accessible to an optimized that you create with an optimization job. You can control access to and from your resources by configuring a VPC. For more information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/infrastructure-give-access.html">Give SageMaker Access to Resources in your Amazon VPC</a>. </p>
- * @public
- */
-export interface OptimizationVpcConfig {
-    /**
-     * <p>The VPC security group IDs, in the form <code>sg-xxxxxxxx</code>. Specify the security groups for the VPC that is specified in the <code>Subnets</code> field.</p>
-     * @public
-     */
-    SecurityGroupIds: string[] | undefined;
-    /**
-     * <p>The ID of the subnets in the VPC to which you want to connect your optimized model.</p>
-     * @public
-     */
-    Subnets: string[] | undefined;
-}
-/**
- * @public
- */
-export interface CreateOptimizationJobRequest {
-    /**
-     * <p>A custom name for the new optimization job.</p>
-     * @public
-     */
-    OptimizationJobName: string | undefined;
-    /**
-     * <p>The Amazon Resource Name (ARN) of an IAM role that enables Amazon SageMaker AI to perform tasks on your behalf. </p> <p>During model optimization, Amazon SageMaker AI needs your permission to:</p> <ul> <li> <p>Read input data from an S3 bucket</p> </li> <li> <p>Write model artifacts to an S3 bucket</p> </li> <li> <p>Write logs to Amazon CloudWatch Logs</p> </li> <li> <p>Publish metrics to Amazon CloudWatch</p> </li> </ul> <p>You grant permissions for all of these tasks to an IAM role. To pass this role to Amazon SageMaker AI, the caller of this API must have the <code>iam:PassRole</code> permission. For more information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-roles.html">Amazon SageMaker AI Roles.</a> </p>
-     * @public
-     */
-    RoleArn: string | undefined;
-    /**
-     * <p>The location of the source model to optimize with an optimization job.</p>
-     * @public
-     */
-    ModelSource: OptimizationJobModelSource | undefined;
-    /**
-     * <p>The type of instance that hosts the optimized model that you create with the optimization job.</p>
-     * @public
-     */
-    DeploymentInstanceType: OptimizationJobDeploymentInstanceType | undefined;
-    /**
-     * <p>The maximum number of instances to use for the optimization job.</p>
-     * @public
-     */
-    MaxInstanceCount?: number | undefined;
-    /**
-     * <p>The environment variables to set in the model container.</p>
-     * @public
-     */
-    OptimizationEnvironment?: Record<string, string> | undefined;
-    /**
-     * <p>Settings for each of the optimization techniques that the job applies.</p>
-     * @public
-     */
-    OptimizationConfigs: OptimizationConfig[] | undefined;
-    /**
-     * <p>Details for where to store the optimized model that you create with the optimization job.</p>
-     * @public
-     */
-    OutputConfig: OptimizationJobOutputConfig | undefined;
-    /**
-     * <p>Specifies a limit to how long a job can run. When the job reaches the time limit, SageMaker ends the job. Use this API to cap costs.</p> <p>To stop a training job, SageMaker sends the algorithm the <code>SIGTERM</code> signal, which delays job termination for 120 seconds. Algorithms can use this 120-second window to save the model artifacts, so the results of training are not lost. </p> <p>The training algorithms provided by SageMaker automatically save the intermediate results of a model training job when possible. This attempt to save artifacts is only a best effort case as model might not be in a state from which it can be saved. For example, if training has just started, the model might not be ready to save. When saved, this intermediate data is a valid model artifact. You can use it to create a model with <code>CreateModel</code>.</p> <note> <p>The Neural Topic Model (NTM) currently does not support saving intermediate model artifacts. When training NTMs, make sure that the maximum runtime is sufficient for the training job to complete.</p> </note>
-     * @public
-     */
-    StoppingCondition: StoppingCondition | undefined;
-    /**
-     * <p>A list of key-value pairs associated with the optimization job. For more information, see <a href="https://docs.aws.amazon.com/general/latest/gr/aws_tagging.html">Tagging Amazon Web Services resources</a> in the <i>Amazon Web Services General Reference Guide</i>.</p>
-     * @public
-     */
-    Tags?: Tag[] | undefined;
-    /**
-     * <p>A VPC in Amazon VPC that your optimized model has access to.</p>
-     * @public
-     */
-    VpcConfig?: OptimizationVpcConfig | undefined;
-}

package/dist-types/models/models_2.d.ts

@@ -1,6 +1,178 @@
-import type { ActionStatus, ActivationState, AIBenchmarkJobStatus, AIRecommendationJobStatus, AlgorithmStatus, AppNetworkAccessType, AppSecurityGroupManagement, AppStatus, AppType, AuthMode, AutoMLJobSecondaryStatus, AutoMLJobStatus, AutoMLProblemTypeConfigName, BatchStrategy, CapacityReservationPreference, CaptureStatus, ClusterNodeProvisioningMode, ClusterNodeRecovery, ClusterStatus, CompilationJobStatus, CustomizationTechnique, DataDistributionType, DeepHealthCheckType, DomainStatus, EdgePackagingJobStatus, EdgePresetDeploymentStatus, EdgePresetDeploymentType, EnabledOrDisabled, EndpointStatus, EvaluationType, FeatureGroupStatus, FeatureStatus, FeatureType, FlowDefinitionStatus, HomeEfsFileSystemCreation, HubContentStatus, HubContentSupportStatus, HubContentType, HubStatus, HumanTaskUiStatus, HyperParameterTuningJobObjectiveType, HyperParameterTuningJobStatus, ImageStatus, ImageVersionStatus, InferenceComponentCapacitySizeType, InputMode, JobType, JoinSource, LastUpdateStatusValue, ObjectiveStatus, OfflineStoreStatusValue, PartnerAppAuthType, PartnerAppType, Peft, ProblemType, ProcessingInstanceType, ProcessingS3CompressionType, ProcessingS3DataDistributionType, ProcessingS3DataType, ProcessingS3InputMode, ProcessingS3UploadMode, Processor, ProductionVariantAcceleratorType, ProductionVariantInstanceType, RecommendationStatus, RedshiftResultCompressionType, RedshiftResultFormat, RetentionType, RuleEvaluationStatus, SchedulerConfigComponent, SchedulerResourceStatus, ServerlessJobType, SharingType, StageStatus, Statistic, StudioLifecycleConfigAppType, TagPropagation, ThroughputMode, TrainingJobStatus, TrialComponentPrimaryStatus, VariantStatus, VendorGuidance, WorkforceIpAddressType } from "./enums";
-import type { ActionSource, AIBenchmarkNetworkConfig, AIBenchmarkOutputResult, AIBenchmarkTarget, AIDatasetConfig, AIModelSource, AIRecommendation, AIRecommendationComputeSpec, AIRecommendationInferenceSpecification, AIRecommendationOutputResult, AIRecommendationPerformanceTarget, AIWorkloadConfigs, AlgorithmSpecification, AlgorithmStatusDetails, AlgorithmValidationSpecification, AppSpecification, ArtifactSource, AsyncInferenceConfig, AthenaDatasetDefinition, AutoMLCandidate, AutoMLChannel, AutoMLComputeConfig, AutoMLDataSplitConfig, AutoMLJobArtifacts, AutoMLJobChannel, AutoMLJobCompletionCriteria, AutoMLJobConfig, AutoMLJobObjective, AutoMLOutputDataConfig, AutoMLPartialFailureReason, AutoMLProblemTypeConfig, AutoMLResolvedAttributes, AutoMLSecurityConfig, AutoRollbackConfig, Autotune, BatchDataCaptureConfig, CfnCreateTemplateProvider, Channel, CheckpointConfig, ClusterAutoScalingConfigOutput, ClusterEventDetail, ClusterInstanceGroupDetails, ClusterNodeDetails, ClusterOrchestrator, ClusterRestrictedInstanceGroupDetails, ClusterTieredStorageConfig, CodeEditorAppImageConfig, CodeRepository, CognitoConfig, CognitoMemberDefinition, CollectionConfiguration, ComputeQuotaConfig, ComputeQuotaTarget, GitConfig, InferenceSpecification, JupyterLabAppImageConfig, KernelGatewayImageConfig, OutputDataConfig, ResourceConfig, ResourceSpec, StoppingCondition, Tag, TransformInput, TransformOutput, TransformResources, UserContext, VpcConfig } from "./models_0";
-import type { ContextSource, DataCaptureConfig, DataQualityAppSpecification, DataQualityBaselineConfig, DataQualityJobInput, DefaultSpaceSettings, DeploymentConfig, DeviceSelectionConfig, DomainSettings, EdgeDeploymentConfig, EdgeDeploymentModelConfig, EdgeOutputConfig, ExplainerConfig, FeatureDefinition, FlowDefinitionOutputConfig, HubS3StorageConfig, HumanLoopActivationConfig, HumanLoopConfig, HumanLoopRequestSource, HyperParameterTrainingJobDefinition, HyperParameterTuningJobConfig, HyperParameterTuningJobWarmStartConfig, InferenceComponentComputeResourceRequirements, InferenceComponentSchedulingConfig, InferenceComponentStartupParameters, InputConfig, JupyterServerAppSettings, KernelGatewayAppSettings, MetadataProperties, MetricsConfig, ModelDeployConfig, MonitoringNetworkConfig, MonitoringOutputConfig, MonitoringResources, MonitoringStoppingCondition, NeoVpcConfig, NetworkConfig, OfflineStoreConfig, OnlineStoreConfig, OutputConfig, ProductionVariant, ProductionVariantManagedInstanceScaling, ProductionVariantRoutingConfig, ProductionVariantServerlessConfig, RetryStrategy, SchedulerConfig, TrainingSpecification, UserSettings } from "./models_1";
+import type { ActionStatus, ActivationState, AIBenchmarkJobStatus, AIRecommendationJobStatus, AlgorithmStatus, AppNetworkAccessType, AppSecurityGroupManagement, AppStatus, AppType, AuthMode, AutoMLJobSecondaryStatus, AutoMLJobStatus, AutoMLProblemTypeConfigName, BatchStrategy, CapacityReservationPreference, CaptureStatus, ClusterNodeProvisioningMode, ClusterNodeRecovery, ClusterStatus, CompilationJobStatus, CustomizationTechnique, DataDistributionType, DeepHealthCheckType, DomainStatus, EdgePackagingJobStatus, EdgePresetDeploymentStatus, EdgePresetDeploymentType, EnabledOrDisabled, EndpointStatus, EvaluationType, FeatureGroupStatus, FeatureStatus, FeatureType, FlowDefinitionStatus, HomeEfsFileSystemCreation, HubContentStatus, HubContentSupportStatus, HubContentType, HubStatus, HumanTaskUiStatus, HyperParameterTuningJobObjectiveType, HyperParameterTuningJobStatus, ImageStatus, ImageVersionStatus, InferenceComponentCapacitySizeType, InputMode, JobType, JoinSource, LastUpdateStatusValue, ObjectiveStatus, OfflineStoreStatusValue, OptimizationJobDeploymentInstanceType, PartnerAppAuthType, PartnerAppType, Peft, ProblemType, ProcessingInstanceType, ProcessingS3CompressionType, ProcessingS3DataDistributionType, ProcessingS3DataType, ProcessingS3InputMode, ProcessingS3UploadMode, Processor, ProductionVariantAcceleratorType, ProductionVariantInstanceType, RecommendationStatus, RedshiftResultCompressionType, RedshiftResultFormat, RetentionType, RuleEvaluationStatus, SchedulerConfigComponent, SchedulerResourceStatus, ServerlessJobType, SharingType, StageStatus, Statistic, StudioLifecycleConfigAppType, TagPropagation, ThroughputMode, TrainingJobStatus, TrialComponentPrimaryStatus, VariantStatus, VendorGuidance, WorkforceIpAddressType } from "./enums";
+import type { ActionSource, AIBenchmarkNetworkConfig, AIBenchmarkOutputResult, AIBenchmarkTarget, AIDatasetConfig, AIModelSource, AIRecommendation, AIRecommendationComputeSpec, AIRecommendationInferenceSpecification, AIRecommendationOutputResult, AIRecommendationPerformanceTarget, AIWorkloadConfigs, AlgorithmSpecification, AlgorithmStatusDetails, AlgorithmValidationSpecification, AppSpecification, ArtifactSource, AsyncInferenceConfig, AthenaDatasetDefinition, AutoMLCandidate, AutoMLChannel, AutoMLComputeConfig, AutoMLDataSplitConfig, AutoMLJobArtifacts, AutoMLJobChannel, AutoMLJobCompletionCriteria, AutoMLJobConfig, AutoMLJobObjective, AutoMLOutputDataConfig, AutoMLPartialFailureReason, AutoMLProblemTypeConfig, AutoMLResolvedAttributes, AutoMLSecurityConfig, AutoRollbackConfig, Autotune, BatchDataCaptureConfig, CfnCreateTemplateProvider, Channel, CheckpointConfig, ClusterAutoScalingConfigOutput, ClusterEventDetail, ClusterInstanceGroupDetails, ClusterNodeDetails, ClusterOrchestrator, ClusterRestrictedInstanceGroupDetails, ClusterRestrictedInstanceGroupsConfigOutput, ClusterTieredStorageConfig, CodeEditorAppImageConfig, CodeRepository, CognitoConfig, CognitoMemberDefinition, CollectionConfiguration, ComputeQuotaConfig, ComputeQuotaTarget, GitConfig, InferenceSpecification, JupyterLabAppImageConfig, KernelGatewayImageConfig, OutputDataConfig, ResourceConfig, ResourceSpec, StoppingCondition, Tag, TransformInput, TransformOutput, TransformResources, UserContext, VpcConfig } from "./models_0";
+import type { ContextSource, DataCaptureConfig, DataQualityAppSpecification, DataQualityBaselineConfig, DataQualityJobInput, DefaultSpaceSettings, DeploymentConfig, DeviceSelectionConfig, DomainSettings, EdgeDeploymentConfig, EdgeDeploymentModelConfig, EdgeOutputConfig, ExplainerConfig, FeatureDefinition, FlowDefinitionOutputConfig, HubS3StorageConfig, HumanLoopActivationConfig, HumanLoopConfig, HumanLoopRequestSource, HyperParameterTrainingJobDefinition, HyperParameterTuningJobConfig, HyperParameterTuningJobWarmStartConfig, InputConfig, JupyterServerAppSettings, KernelGatewayAppSettings, MetadataProperties, MetricsConfig, ModelCompilationConfig, ModelDeployConfig, ModelQuantizationConfig, ModelShardingConfig, ModelSpeculativeDecodingConfig, MonitoringNetworkConfig, MonitoringOutputConfig, MonitoringResources, MonitoringStoppingCondition, NeoVpcConfig, NetworkConfig, OfflineStoreConfig, OnlineStoreConfig, OptimizationJobModelSource, OptimizationSageMakerModel, OutputConfig, ProductionVariant, ProductionVariantManagedInstanceScaling, ProductionVariantRoutingConfig, ProductionVariantServerlessConfig, RetryStrategy, SchedulerConfig, TrainingSpecification, UserSettings } from "./models_1";
+/**
+ * <p>Settings for an optimization technique that you apply with a model optimization job.</p>
+ * @public
+ */
+export type OptimizationConfig = OptimizationConfig.ModelCompilationConfigMember | OptimizationConfig.ModelQuantizationConfigMember | OptimizationConfig.ModelShardingConfigMember | OptimizationConfig.ModelSpeculativeDecodingConfigMember | OptimizationConfig.$UnknownMember;
+/**
+ * @public
+ */
+export declare namespace OptimizationConfig {
+    /**
+     * <p>Settings for the model quantization technique that's applied by a model optimization job.</p>
+     * @public
+     */
+    interface ModelQuantizationConfigMember {
+        ModelQuantizationConfig: ModelQuantizationConfig;
+        ModelCompilationConfig?: never;
+        ModelShardingConfig?: never;
+        ModelSpeculativeDecodingConfig?: never;
+        $unknown?: never;
+    }
+    /**
+     * <p>Settings for the model compilation technique that's applied by a model optimization job.</p>
+     * @public
+     */
+    interface ModelCompilationConfigMember {
+        ModelQuantizationConfig?: never;
+        ModelCompilationConfig: ModelCompilationConfig;
+        ModelShardingConfig?: never;
+        ModelSpeculativeDecodingConfig?: never;
+        $unknown?: never;
+    }
+    /**
+     * <p>Settings for the model sharding technique that's applied by a model optimization job.</p>
+     * @public
+     */
+    interface ModelShardingConfigMember {
+        ModelQuantizationConfig?: never;
+        ModelCompilationConfig?: never;
+        ModelShardingConfig: ModelShardingConfig;
+        ModelSpeculativeDecodingConfig?: never;
+        $unknown?: never;
+    }
+    /**
+     * <p>Settings for the model speculative decoding technique that's applied by a model optimization job.</p>
+     * @public
+     */
+    interface ModelSpeculativeDecodingConfigMember {
+        ModelQuantizationConfig?: never;
+        ModelCompilationConfig?: never;
+        ModelShardingConfig?: never;
+        ModelSpeculativeDecodingConfig: ModelSpeculativeDecodingConfig;
+        $unknown?: never;
+    }
+    /**
+     * @public
+     */
+    interface $UnknownMember {
+        ModelQuantizationConfig?: never;
+        ModelCompilationConfig?: never;
+        ModelShardingConfig?: never;
+        ModelSpeculativeDecodingConfig?: never;
+        $unknown: [string, any];
+    }
+    /**
+     * @deprecated unused in schema-serde mode.
+     *
+     */
+    interface Visitor<T> {
+        ModelQuantizationConfig: (value: ModelQuantizationConfig) => T;
+        ModelCompilationConfig: (value: ModelCompilationConfig) => T;
+        ModelShardingConfig: (value: ModelShardingConfig) => T;
+        ModelSpeculativeDecodingConfig: (value: ModelSpeculativeDecodingConfig) => T;
+        _: (name: string, value: any) => T;
+    }
+}
+/**
+ * <p>Details for where to store the optimized model that you create with the optimization job.</p>
+ * @public
+ */
+export interface OptimizationJobOutputConfig {
+    /**
+     * <p>The Amazon Resource Name (ARN) of a key in Amazon Web Services KMS. SageMaker uses they key to encrypt the artifacts of the optimized model when SageMaker uploads the model to Amazon S3.</p>
+     * @public
+     */
+    KmsKeyId?: string | undefined;
+    /**
+     * <p>The Amazon S3 URI for where to store the optimized model that you create with an optimization job.</p>
+     * @public
+     */
+    S3OutputLocation: string | undefined;
+    /**
+     * <p>The name of a SageMaker model to use as the output destination for an optimization job.</p>
+     * @public
+     */
+    SageMakerModel?: OptimizationSageMakerModel | undefined;
+}
+/**
+ * <p>A VPC in Amazon VPC that's accessible to an optimized that you create with an optimization job. You can control access to and from your resources by configuring a VPC. For more information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/infrastructure-give-access.html">Give SageMaker Access to Resources in your Amazon VPC</a>. </p>
+ * @public
+ */
+export interface OptimizationVpcConfig {
+    /**
+     * <p>The VPC security group IDs, in the form <code>sg-xxxxxxxx</code>. Specify the security groups for the VPC that is specified in the <code>Subnets</code> field.</p>
+     * @public
+     */
+    SecurityGroupIds: string[] | undefined;
+    /**
+     * <p>The ID of the subnets in the VPC to which you want to connect your optimized model.</p>
+     * @public
+     */
+    Subnets: string[] | undefined;
+}
+/**
+ * @public
+ */
+export interface CreateOptimizationJobRequest {
+    /**
+     * <p>A custom name for the new optimization job.</p>
+     * @public
+     */
+    OptimizationJobName: string | undefined;
+    /**
+     * <p>The Amazon Resource Name (ARN) of an IAM role that enables Amazon SageMaker AI to perform tasks on your behalf. </p> <p>During model optimization, Amazon SageMaker AI needs your permission to:</p> <ul> <li> <p>Read input data from an S3 bucket</p> </li> <li> <p>Write model artifacts to an S3 bucket</p> </li> <li> <p>Write logs to Amazon CloudWatch Logs</p> </li> <li> <p>Publish metrics to Amazon CloudWatch</p> </li> </ul> <p>You grant permissions for all of these tasks to an IAM role. To pass this role to Amazon SageMaker AI, the caller of this API must have the <code>iam:PassRole</code> permission. For more information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-roles.html">Amazon SageMaker AI Roles.</a> </p>
+     * @public
+     */
+    RoleArn: string | undefined;
+    /**
+     * <p>The location of the source model to optimize with an optimization job.</p>
+     * @public
+     */
+    ModelSource: OptimizationJobModelSource | undefined;
+    /**
+     * <p>The type of instance that hosts the optimized model that you create with the optimization job.</p>
+     * @public
+     */
+    DeploymentInstanceType: OptimizationJobDeploymentInstanceType | undefined;
+    /**
+     * <p>The maximum number of instances to use for the optimization job.</p>
+     * @public
+     */
+    MaxInstanceCount?: number | undefined;
+    /**
+     * <p>The environment variables to set in the model container.</p>
+     * @public
+     */
+    OptimizationEnvironment?: Record<string, string> | undefined;
+    /**
+     * <p>Settings for each of the optimization techniques that the job applies.</p>
+     * @public
+     */
+    OptimizationConfigs: OptimizationConfig[] | undefined;
+    /**
+     * <p>Details for where to store the optimized model that you create with the optimization job.</p>
+     * @public
+     */
+    OutputConfig: OptimizationJobOutputConfig | undefined;
+    /**
+     * <p>Specifies a limit to how long a job can run. When the job reaches the time limit, SageMaker ends the job. Use this API to cap costs.</p> <p>To stop a training job, SageMaker sends the algorithm the <code>SIGTERM</code> signal, which delays job termination for 120 seconds. Algorithms can use this 120-second window to save the model artifacts, so the results of training are not lost. </p> <p>The training algorithms provided by SageMaker automatically save the intermediate results of a model training job when possible. This attempt to save artifacts is only a best effort case as model might not be in a state from which it can be saved. For example, if training has just started, the model might not be ready to save. When saved, this intermediate data is a valid model artifact. You can use it to create a model with <code>CreateModel</code>.</p> <note> <p>The Neural Topic Model (NTM) currently does not support saving intermediate model artifacts. When training NTMs, make sure that the maximum runtime is sufficient for the training job to complete.</p> </note>
+     * @public
+     */
+    StoppingCondition: StoppingCondition | undefined;
+    /**
+     * <p>A list of key-value pairs associated with the optimization job. For more information, see <a href="https://docs.aws.amazon.com/general/latest/gr/aws_tagging.html">Tagging Amazon Web Services resources</a> in the <i>Amazon Web Services General Reference Guide</i>.</p>
+     * @public
+     */
+    Tags?: Tag[] | undefined;
+    /**
+     * <p>A VPC in Amazon VPC that your optimized model has access to.</p>
+     * @public
+     */
+    VpcConfig?: OptimizationVpcConfig | undefined;
+}
 /**
  * @public
  */
@@ -4358,6 +4530,11 @@ export interface DescribeClusterResponse {
      * @public
      */
     RestrictedInstanceGroups?: ClusterRestrictedInstanceGroupDetails[] | undefined;
+    /**
+     * <p>The configuration for the restricted instance groups (RIG) in the SageMaker HyperPod cluster.</p>
+     * @public
+     */
+    RestrictedInstanceGroupsConfig?: ClusterRestrictedInstanceGroupsConfigOutput | undefined;
     /**
      * <p>Specifies an Amazon Virtual Private Cloud (VPC) that your SageMaker jobs, hosted models, and compute resources have access to. You can control access to and from your resources by configuring a VPC. For more information, see <a href="https://docs.aws.amazon.com/sagemaker/latest/dg/infrastructure-give-access.html">Give SageMaker Access to Resources in your Amazon VPC</a>. </p>
      * @public
@@ -7044,102 +7221,3 @@ export interface InferenceComponentPlacementStatus {
      */
     CurrentCopyCount: number | undefined;
 }
-/**
- * <p>Details about the runtime settings for the model that is deployed with the inference component.</p>
- * @public
- */
-export interface InferenceComponentRuntimeConfigSummary {
-    /**
-     * <p>The number of runtime copies of the model container that you requested to deploy with the inference component.</p>
-     * @public
-     */
-    DesiredCopyCount?: number | undefined;
-    /**
-     * <p>The number of runtime copies of the model container that are currently deployed.</p>
-     * @public
-     */
-    CurrentCopyCount?: number | undefined;
-    /**
-     * <p>The placement status of the inference component across instance types. Shows how the inference component copies are distributed across instance types.</p>
-     * @public
-     */
-    PlacementStatus?: InferenceComponentPlacementStatus[] | undefined;
-}
-/**
- * <p>Details about the resources that are deployed with this inference component.</p>
- * @public
- */
-export interface InferenceComponentContainerSpecificationSummary {
-    /**
-     * <p>Gets the Amazon EC2 Container Registry path of the docker image of the model that is hosted in this <a href="https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ProductionVariant.html">ProductionVariant</a>.</p> <p>If you used the <code>registry/repository[:tag]</code> form to specify the image path of the primary container when you created the model hosted in this <code>ProductionVariant</code>, the path resolves to a path of the form <code>registry/repository[@digest]</code>. A digest is a hash value that identifies a specific version of an image. For information about Amazon ECR paths, see <a href="https://docs.aws.amazon.com/AmazonECR/latest/userguide/docker-pull-ecr-image.html">Pulling an Image</a> in the <i>Amazon ECR User Guide</i>.</p>
-     * @public
-     */
-    DeployedImage?: DeployedImage | undefined;
-    /**
-     * <p>The Amazon S3 path where the model artifacts are stored.</p>
-     * @public
-     */
-    ArtifactUrl?: string | undefined;
-    /**
-     * <p>The environment variables to set in the Docker container.</p>
-     * @public
-     */
-    Environment?: Record<string, string> | undefined;
-}
-/**
- * <p>Settings that affect how the inference component caches data.</p>
- * @public
- */
-export interface InferenceComponentDataCacheConfigSummary {
-    /**
-     * <p>Indicates whether the inference component caches model artifacts as part of the auto scaling process.</p>
-     * @public
-     */
-    EnableCaching: boolean | undefined;
-}
-/**
- * <p>Details about the resources that are deployed with this inference component.</p>
- * @public
- */
-export interface InferenceComponentSpecificationSummary {
-    /**
-     * <p>The ML compute instance type associated with this inference component specification.</p>
-     * @public
-     */
-    InstanceType?: ProductionVariantInstanceType | undefined;
-    /**
-     * <p>The name of the SageMaker AI model object that is deployed with the inference component.</p>
-     * @public
-     */
-    ModelName?: string | undefined;
-    /**
-     * <p>Details about the container that provides the runtime environment for the model that is deployed with the inference component.</p>
-     * @public
-     */
-    Container?: InferenceComponentContainerSpecificationSummary | undefined;
-    /**
-     * <p>Settings that take effect while the model container starts up.</p>
-     * @public
-     */
-    StartupParameters?: InferenceComponentStartupParameters | undefined;
-    /**
-     * <p>The compute resources allocated to run the model, plus any adapter models, that you assign to the inference component.</p>
-     * @public
-     */
-    ComputeResourceRequirements?: InferenceComponentComputeResourceRequirements | undefined;
-    /**
-     * <p>The name of the base inference component that contains this inference component.</p>
-     * @public
-     */
-    BaseInferenceComponentName?: string | undefined;
-    /**
-     * <p>Settings that affect how the inference component caches data.</p>
-     * @public
-     */
-    DataCacheConfig?: InferenceComponentDataCacheConfigSummary | undefined;
-    /**
-     * <p>The scheduling configuration that determines how inference component copies are placed across available instances when copies are added or removed.</p>
-     * @public
-     */
-    SchedulingConfig?: InferenceComponentSchedulingConfig | undefined;
-}