sagemaker-core 1.0.62__py3-none-any.whl → 2.3.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (358) hide show
  1. sagemaker/__init__.py +2 -0
  2. sagemaker/core/__init__.py +16 -0
  3. sagemaker/core/_studio.py +116 -0
  4. sagemaker/core/_version.py +11 -0
  5. sagemaker/core/accept_types.py +131 -0
  6. sagemaker/core/analytics.py +744 -0
  7. sagemaker/core/apiutils/__init__.py +13 -0
  8. sagemaker/core/apiutils/_base_types.py +228 -0
  9. sagemaker/core/apiutils/_boto_functions.py +130 -0
  10. sagemaker/core/apiutils/_utils.py +34 -0
  11. sagemaker/core/base_deserializers.py +35 -0
  12. sagemaker/core/base_serializers.py +35 -0
  13. sagemaker/core/clarify/__init__.py +2898 -0
  14. sagemaker/core/collection.py +467 -0
  15. sagemaker/core/common_utils.py +2399 -0
  16. sagemaker/core/compute_resource_requirements/__init__.py +18 -0
  17. sagemaker/core/compute_resource_requirements/resource_requirements.py +94 -0
  18. sagemaker/core/config/__init__.py +181 -0
  19. sagemaker/core/config/config.py +238 -0
  20. sagemaker/core/config/config_manager.py +595 -0
  21. sagemaker/core/config/config_schema.py +1220 -0
  22. sagemaker/core/config/config_utils.py +297 -0
  23. {sagemaker_core/main → sagemaker/core}/config_schema.py +408 -3
  24. sagemaker/core/constants.py +73 -0
  25. sagemaker/core/content_types.py +137 -0
  26. sagemaker/core/debugger/__init__.py +39 -0
  27. sagemaker/core/debugger/debugger.py +945 -0
  28. sagemaker/core/debugger/framework_profile.py +292 -0
  29. sagemaker/core/debugger/metrics_config.py +468 -0
  30. sagemaker/core/debugger/profiler.py +42 -0
  31. sagemaker/core/debugger/profiler_config.py +190 -0
  32. sagemaker/core/debugger/profiler_constants.py +40 -0
  33. sagemaker/core/debugger/utils.py +148 -0
  34. sagemaker/core/deprecations.py +254 -0
  35. sagemaker/core/deserializers/__init__.py +10 -0
  36. sagemaker/core/deserializers/base.py +424 -0
  37. sagemaker/core/deserializers/implementations.py +157 -0
  38. sagemaker/core/drift_check_baselines.py +106 -0
  39. sagemaker/core/enums.py +51 -0
  40. sagemaker/core/environment_variables.py +101 -0
  41. sagemaker/core/exceptions.py +108 -0
  42. sagemaker/core/experiments/__init__.py +53 -0
  43. sagemaker/core/experiments/_api_types.py +251 -0
  44. sagemaker/core/experiments/_environment.py +124 -0
  45. sagemaker/core/experiments/_helper.py +294 -0
  46. sagemaker/core/experiments/_metrics.py +333 -0
  47. sagemaker/core/experiments/_run_context.py +58 -0
  48. sagemaker/core/experiments/_utils.py +216 -0
  49. sagemaker/core/experiments/experiment.py +247 -0
  50. sagemaker/core/experiments/run.py +970 -0
  51. sagemaker/core/experiments/trial.py +296 -0
  52. sagemaker/core/experiments/trial_component.py +387 -0
  53. sagemaker/core/explainer/__init__.py +24 -0
  54. sagemaker/core/explainer/clarify_explainer_config.py +298 -0
  55. sagemaker/core/explainer/explainer_config.py +44 -0
  56. sagemaker/core/fw_utils.py +1220 -0
  57. sagemaker/core/git_utils.py +415 -0
  58. sagemaker/core/helper/pipeline_variable.py +82 -0
  59. sagemaker/core/helper/session_helper.py +2977 -0
  60. sagemaker/core/hyperparameters.py +172 -0
  61. sagemaker/core/image_retriever/__init__.py +3 -0
  62. sagemaker/core/image_retriever/image_retriever.py +640 -0
  63. sagemaker/core/image_retriever/image_retriever_utils.py +509 -0
  64. sagemaker/core/image_retriever/test.py +7 -0
  65. sagemaker/core/image_uri_config/autogluon.json +1335 -0
  66. sagemaker/core/image_uri_config/blazingtext.json +50 -0
  67. sagemaker/core/image_uri_config/chainer.json +104 -0
  68. sagemaker/core/image_uri_config/clarify.json +39 -0
  69. sagemaker/core/image_uri_config/coach-mxnet.json +70 -0
  70. sagemaker/core/image_uri_config/coach-tensorflow.json +186 -0
  71. sagemaker/core/image_uri_config/data-wrangler.json +91 -0
  72. sagemaker/core/image_uri_config/debugger.json +34 -0
  73. sagemaker/core/image_uri_config/detailed-profiler.json +18 -0
  74. sagemaker/core/image_uri_config/djl-deepspeed.json +385 -0
  75. sagemaker/core/image_uri_config/djl-fastertransformer.json +167 -0
  76. sagemaker/core/image_uri_config/djl-lmi.json +136 -0
  77. sagemaker/core/image_uri_config/djl-neuronx.json +258 -0
  78. sagemaker/core/image_uri_config/djl-tensorrtllm.json +262 -0
  79. sagemaker/core/image_uri_config/factorization-machines.json +50 -0
  80. sagemaker/core/image_uri_config/forecasting-deepar.json +50 -0
  81. sagemaker/core/image_uri_config/huggingface-llm-neuronx.json +770 -0
  82. sagemaker/core/image_uri_config/huggingface-llm.json +1267 -0
  83. sagemaker/core/image_uri_config/huggingface-neuron.json +52 -0
  84. sagemaker/core/image_uri_config/huggingface-neuronx.json +686 -0
  85. sagemaker/core/image_uri_config/huggingface-tei-cpu.json +298 -0
  86. sagemaker/core/image_uri_config/huggingface-tei.json +298 -0
  87. sagemaker/core/image_uri_config/huggingface-training-compiler.json +195 -0
  88. sagemaker/core/image_uri_config/huggingface-vllm-neuronx.json +38 -0
  89. sagemaker/core/image_uri_config/huggingface.json +2287 -0
  90. sagemaker/core/image_uri_config/hyperpod-recipes-neuron.json +52 -0
  91. sagemaker/core/image_uri_config/image-classification-neo.json +43 -0
  92. sagemaker/core/image_uri_config/image-classification.json +50 -0
  93. sagemaker/core/image_uri_config/inferentia-mxnet.json +88 -0
  94. sagemaker/core/image_uri_config/inferentia-pytorch.json +127 -0
  95. sagemaker/core/image_uri_config/inferentia-tensorflow.json +88 -0
  96. sagemaker/core/image_uri_config/instance_gpu_info.json +782 -0
  97. sagemaker/core/image_uri_config/ipinsights.json +50 -0
  98. sagemaker/core/image_uri_config/kmeans.json +50 -0
  99. sagemaker/core/image_uri_config/knn.json +50 -0
  100. sagemaker/core/image_uri_config/lda.json +26 -0
  101. sagemaker/core/image_uri_config/linear-learner.json +50 -0
  102. sagemaker/core/image_uri_config/model-monitor.json +42 -0
  103. sagemaker/core/image_uri_config/mxnet.json +1154 -0
  104. sagemaker/core/image_uri_config/neo-mxnet.json +64 -0
  105. sagemaker/core/image_uri_config/neo-pytorch.json +341 -0
  106. sagemaker/core/image_uri_config/neo-tensorflow.json +109 -0
  107. sagemaker/core/image_uri_config/ntm.json +50 -0
  108. sagemaker/core/image_uri_config/object-detection.json +50 -0
  109. sagemaker/core/image_uri_config/object2vec.json +50 -0
  110. sagemaker/core/image_uri_config/pca.json +50 -0
  111. sagemaker/core/image_uri_config/pytorch-neuron.json +43 -0
  112. sagemaker/core/image_uri_config/pytorch-smp.json +218 -0
  113. sagemaker/core/image_uri_config/pytorch-training-compiler.json +80 -0
  114. sagemaker/core/image_uri_config/pytorch.json +3101 -0
  115. sagemaker/core/image_uri_config/randomcutforest.json +50 -0
  116. sagemaker/core/image_uri_config/ray-pytorch.json +46 -0
  117. sagemaker/core/image_uri_config/ray-tensorflow.json +194 -0
  118. sagemaker/core/image_uri_config/sagemaker-base-python.json +46 -0
  119. sagemaker/core/image_uri_config/sagemaker-distribution.json +37 -0
  120. sagemaker/core/image_uri_config/sagemaker-geospatial.json +13 -0
  121. sagemaker/core/image_uri_config/sagemaker-tritonserver.json +252 -0
  122. sagemaker/core/image_uri_config/semantic-segmentation.json +50 -0
  123. sagemaker/core/image_uri_config/seq2seq.json +50 -0
  124. sagemaker/core/image_uri_config/sklearn.json +494 -0
  125. sagemaker/core/image_uri_config/spark.json +280 -0
  126. sagemaker/core/image_uri_config/sparkml-serving.json +97 -0
  127. sagemaker/core/image_uri_config/stabilityai.json +53 -0
  128. sagemaker/core/image_uri_config/tensorflow.json +5086 -0
  129. sagemaker/core/image_uri_config/vw.json +25 -0
  130. sagemaker/core/image_uri_config/xgboost-neo.json +43 -0
  131. sagemaker/core/image_uri_config/xgboost.json +972 -0
  132. sagemaker/core/image_uris.py +816 -0
  133. sagemaker/core/inference_config.py +144 -0
  134. sagemaker/core/inference_recommender/__init__.py +18 -0
  135. sagemaker/core/inference_recommender/inference_recommender_mixin.py +622 -0
  136. sagemaker/core/inputs.py +366 -0
  137. sagemaker/core/instance_group.py +61 -0
  138. sagemaker/core/instance_types.py +164 -0
  139. sagemaker/core/instance_types_gpu_info.py +43 -0
  140. sagemaker/core/interactive_apps/__init__.py +41 -0
  141. sagemaker/core/interactive_apps/base_interactive_app.py +204 -0
  142. sagemaker/core/interactive_apps/detail_profiler_app.py +139 -0
  143. sagemaker/core/interactive_apps/tensorboard.py +149 -0
  144. sagemaker/core/iterators.py +197 -0
  145. sagemaker/core/job.py +380 -0
  146. sagemaker/core/jumpstart/__init__.py +156 -0
  147. sagemaker/core/jumpstart/accessors.py +390 -0
  148. sagemaker/core/jumpstart/artifacts/__init__.py +69 -0
  149. sagemaker/core/jumpstart/artifacts/environment_variables.py +252 -0
  150. sagemaker/core/jumpstart/artifacts/hyperparameters.py +120 -0
  151. sagemaker/core/jumpstart/artifacts/image_uris.py +139 -0
  152. sagemaker/core/jumpstart/artifacts/incremental_training.py +87 -0
  153. sagemaker/core/jumpstart/artifacts/instance_types.py +223 -0
  154. sagemaker/core/jumpstart/artifacts/kwargs.py +289 -0
  155. sagemaker/core/jumpstart/artifacts/metric_definitions.py +117 -0
  156. sagemaker/core/jumpstart/artifacts/model_packages.py +202 -0
  157. sagemaker/core/jumpstart/artifacts/model_uris.py +252 -0
  158. sagemaker/core/jumpstart/artifacts/payloads.py +96 -0
  159. sagemaker/core/jumpstart/artifacts/predictors.py +540 -0
  160. sagemaker/core/jumpstart/artifacts/resource_names.py +86 -0
  161. sagemaker/core/jumpstart/artifacts/resource_requirements.py +162 -0
  162. sagemaker/core/jumpstart/artifacts/script_uris.py +172 -0
  163. sagemaker/core/jumpstart/cache.py +663 -0
  164. sagemaker/core/jumpstart/configs.py +50 -0
  165. sagemaker/core/jumpstart/constants.py +198 -0
  166. sagemaker/core/jumpstart/deserializers.py +81 -0
  167. sagemaker/core/jumpstart/document.py +76 -0
  168. sagemaker/core/jumpstart/enums.py +168 -0
  169. sagemaker/core/jumpstart/exceptions.py +236 -0
  170. sagemaker/core/jumpstart/factory/utils.py +833 -0
  171. sagemaker/core/jumpstart/filters.py +597 -0
  172. sagemaker/core/jumpstart/hub/constants.py +16 -0
  173. sagemaker/core/jumpstart/hub/hub.py +291 -0
  174. sagemaker/core/jumpstart/hub/interfaces.py +936 -0
  175. sagemaker/core/jumpstart/hub/parser_utils.py +70 -0
  176. sagemaker/core/jumpstart/hub/parsers.py +288 -0
  177. sagemaker/core/jumpstart/hub/types.py +35 -0
  178. sagemaker/core/jumpstart/hub/utils.py +260 -0
  179. sagemaker/core/jumpstart/models.py +501 -0
  180. sagemaker/core/jumpstart/notebook_utils.py +575 -0
  181. sagemaker/core/jumpstart/parameters.py +20 -0
  182. sagemaker/core/jumpstart/payload_utils.py +239 -0
  183. sagemaker/core/jumpstart/region_config.json +171 -0
  184. sagemaker/core/jumpstart/search.py +171 -0
  185. sagemaker/core/jumpstart/serializers.py +81 -0
  186. sagemaker/core/jumpstart/session_utils.py +234 -0
  187. sagemaker/core/jumpstart/types.py +3044 -0
  188. sagemaker/core/jumpstart/utils.py +1731 -0
  189. sagemaker/core/jumpstart/validators.py +257 -0
  190. sagemaker/core/lambda_helper.py +312 -0
  191. sagemaker/core/lineage/__init__.py +42 -0
  192. sagemaker/core/lineage/_api_types.py +239 -0
  193. sagemaker/core/lineage/_utils.py +49 -0
  194. sagemaker/core/lineage/action.py +345 -0
  195. sagemaker/core/lineage/artifact.py +646 -0
  196. sagemaker/core/lineage/association.py +190 -0
  197. sagemaker/core/lineage/context.py +505 -0
  198. sagemaker/core/lineage/lineage_trial_component.py +191 -0
  199. sagemaker/core/lineage/query.py +732 -0
  200. sagemaker/core/lineage/visualizer.py +346 -0
  201. sagemaker/core/local/__init__.py +18 -0
  202. sagemaker/core/local/data.py +423 -0
  203. sagemaker/core/local/entities.py +678 -0
  204. sagemaker/core/local/exceptions.py +17 -0
  205. sagemaker/core/local/image.py +1243 -0
  206. sagemaker/core/local/local_session.py +739 -0
  207. sagemaker/core/local/utils.py +246 -0
  208. sagemaker/core/logs.py +181 -0
  209. sagemaker/core/metadata_properties.py +56 -0
  210. sagemaker/core/metric_definitions.py +91 -0
  211. sagemaker/core/mlflow/__init__.py +38 -0
  212. sagemaker/core/mlflow/forward_sagemaker_metrics.py +44 -0
  213. sagemaker/core/model_card/__init__.py +26 -0
  214. sagemaker/core/model_life_cycle.py +51 -0
  215. sagemaker/core/model_metrics.py +160 -0
  216. sagemaker/core/model_monitor/__init__.py +66 -0
  217. sagemaker/core/model_monitor/clarify_model_monitoring.py +1497 -0
  218. sagemaker/core/model_monitor/cron_expression_generator.py +82 -0
  219. sagemaker/core/model_monitor/data_capture_config.py +115 -0
  220. sagemaker/core/model_monitor/data_quality_monitoring_config.py +66 -0
  221. sagemaker/core/model_monitor/dataset_format.py +102 -0
  222. sagemaker/core/model_monitor/model_monitoring.py +4266 -0
  223. sagemaker/core/model_monitor/monitoring_alert.py +76 -0
  224. sagemaker/core/model_monitor/monitoring_files.py +506 -0
  225. sagemaker/core/model_monitor/utils.py +793 -0
  226. sagemaker/core/model_registry.py +480 -0
  227. sagemaker/core/model_uris.py +97 -0
  228. sagemaker/core/modules/__init__.py +19 -0
  229. sagemaker/core/modules/configs.py +239 -0
  230. sagemaker/core/modules/constants.py +37 -0
  231. sagemaker/core/modules/distributed.py +182 -0
  232. sagemaker/core/modules/local_core/local_container.py +605 -0
  233. sagemaker/core/modules/templates.py +83 -0
  234. sagemaker/core/modules/train/__init__.py +14 -0
  235. sagemaker/core/modules/train/container_drivers/__init__.py +14 -0
  236. sagemaker/core/modules/train/container_drivers/common/__init__.py +14 -0
  237. sagemaker/core/modules/train/container_drivers/common/utils.py +205 -0
  238. sagemaker/core/modules/train/container_drivers/distributed_drivers/__init__.py +14 -0
  239. sagemaker/core/modules/train/container_drivers/distributed_drivers/basic_script_driver.py +81 -0
  240. sagemaker/core/modules/train/container_drivers/distributed_drivers/mpi_driver.py +123 -0
  241. sagemaker/core/modules/train/container_drivers/distributed_drivers/mpi_utils.py +302 -0
  242. sagemaker/core/modules/train/container_drivers/distributed_drivers/torchrun_driver.py +129 -0
  243. sagemaker/core/modules/train/container_drivers/scripts/__init__.py +14 -0
  244. sagemaker/core/modules/train/container_drivers/scripts/environment.py +305 -0
  245. sagemaker/core/modules/train/sm_recipes/__init__.py +0 -0
  246. sagemaker/core/modules/train/sm_recipes/utils.py +330 -0
  247. sagemaker/core/modules/types.py +19 -0
  248. sagemaker/core/modules/utils.py +194 -0
  249. sagemaker/core/network.py +185 -0
  250. sagemaker/core/parameter.py +173 -0
  251. sagemaker/core/payloads.py +185 -0
  252. sagemaker/core/processing.py +1599 -0
  253. sagemaker/core/remote_function/__init__.py +19 -0
  254. sagemaker/core/remote_function/checkpoint_location.py +47 -0
  255. sagemaker/core/remote_function/client.py +1310 -0
  256. sagemaker/core/remote_function/core/__init__.py +0 -0
  257. sagemaker/core/remote_function/core/_custom_dispatch_table.py +72 -0
  258. sagemaker/core/remote_function/core/pipeline_variables.py +347 -0
  259. sagemaker/core/remote_function/core/serialization.py +410 -0
  260. sagemaker/core/remote_function/core/stored_function.py +223 -0
  261. sagemaker/core/remote_function/custom_file_filter.py +128 -0
  262. sagemaker/core/remote_function/errors.py +102 -0
  263. sagemaker/core/remote_function/invoke_function.py +167 -0
  264. sagemaker/core/remote_function/job.py +2121 -0
  265. sagemaker/core/remote_function/logging_config.py +38 -0
  266. sagemaker/core/remote_function/runtime_environment/__init__.py +14 -0
  267. sagemaker/core/remote_function/runtime_environment/bootstrap_runtime_environment.py +605 -0
  268. sagemaker/core/remote_function/runtime_environment/mpi_utils_remote.py +252 -0
  269. sagemaker/core/remote_function/runtime_environment/runtime_environment_manager.py +554 -0
  270. sagemaker/core/remote_function/runtime_environment/spark_app.py +18 -0
  271. sagemaker/core/remote_function/spark_config.py +149 -0
  272. sagemaker/core/resource_requirements.py +168 -0
  273. {sagemaker_core/main → sagemaker/core}/resources.py +19098 -10895
  274. sagemaker/core/s3/__init__.py +41 -0
  275. sagemaker/core/s3/client.py +367 -0
  276. sagemaker/core/s3/utils.py +175 -0
  277. sagemaker/core/script_uris.py +93 -0
  278. sagemaker/core/serializers/__init__.py +11 -0
  279. sagemaker/core/serializers/base.py +510 -0
  280. sagemaker/core/serializers/implementations.py +159 -0
  281. sagemaker/core/serializers/utils.py +223 -0
  282. sagemaker/core/serverless_inference_config.py +63 -0
  283. sagemaker/core/session_settings.py +55 -0
  284. sagemaker/core/shapes/__init__.py +3 -0
  285. sagemaker/core/shapes/model_card_shapes.py +159 -0
  286. {sagemaker_core/main → sagemaker/core/shapes}/shapes.py +5810 -1806
  287. sagemaker/core/spark/__init__.py +16 -0
  288. sagemaker/core/spark/defaults.py +16 -0
  289. sagemaker/core/spark/processing.py +1380 -0
  290. sagemaker/core/telemetry/__init__.py +23 -0
  291. sagemaker/core/telemetry/constants.py +82 -0
  292. sagemaker/core/telemetry/telemetry_logging.py +285 -0
  293. sagemaker/core/tools/__init__.py +1 -0
  294. {sagemaker_core → sagemaker/core}/tools/codegen.py +4 -4
  295. {sagemaker_core → sagemaker/core}/tools/constants.py +23 -15
  296. {sagemaker_core → sagemaker/core}/tools/data_extractor.py +1 -1
  297. {sagemaker_core → sagemaker/core}/tools/method.py +1 -1
  298. sagemaker/core/tools/model_card/generate_model_card_from_schema.py +562 -0
  299. {sagemaker_core → sagemaker/core}/tools/resources_codegen.py +165 -98
  300. {sagemaker_core → sagemaker/core}/tools/resources_extractor.py +5 -13
  301. {sagemaker_core → sagemaker/core}/tools/shapes_codegen.py +16 -17
  302. {sagemaker_core → sagemaker/core}/tools/shapes_extractor.py +29 -67
  303. {sagemaker_core → sagemaker/core}/tools/templates.py +39 -17
  304. sagemaker/core/training/__init__.py +14 -0
  305. sagemaker/core/training/configs.py +345 -0
  306. sagemaker/core/training/constants.py +37 -0
  307. sagemaker/core/training/utils.py +77 -0
  308. sagemaker/core/training_compiler/__init__.py +16 -0
  309. sagemaker/core/training_compiler/config.py +197 -0
  310. sagemaker/core/training_compiler_config.py +197 -0
  311. sagemaker/core/transformer.py +793 -0
  312. sagemaker/core/user_agent.py +76 -0
  313. sagemaker/core/utilities/__init__.py +24 -0
  314. sagemaker/core/utilities/cache.py +169 -0
  315. sagemaker/core/utilities/search_expression.py +133 -0
  316. sagemaker/core/utils/__init__.py +48 -0
  317. sagemaker/core/utils/code_injection/__init__.py +0 -0
  318. {sagemaker_core/main → sagemaker/core/utils}/code_injection/codec.py +2 -2
  319. {sagemaker_core/main → sagemaker/core/utils}/code_injection/shape_dag.py +5979 -176
  320. {sagemaker_core/main → sagemaker/core/utils}/exceptions.py +8 -8
  321. sagemaker_core/main/default_configs_helper.py → sagemaker/core/utils/intelligent_defaults_helper.py +5 -6
  322. {sagemaker_core/main → sagemaker/core/utils}/logs.py +1 -2
  323. {sagemaker_core/main → sagemaker/core/utils}/utils.py +27 -22
  324. sagemaker/core/workflow/__init__.py +152 -0
  325. sagemaker/core/workflow/conditions.py +313 -0
  326. sagemaker/core/workflow/entities.py +58 -0
  327. sagemaker/core/workflow/execution_variables.py +89 -0
  328. sagemaker/core/workflow/functions.py +193 -0
  329. sagemaker/core/workflow/parameters.py +222 -0
  330. sagemaker/core/workflow/pipeline_context.py +394 -0
  331. sagemaker/core/workflow/pipeline_definition_config.py +31 -0
  332. sagemaker/core/workflow/properties.py +285 -0
  333. sagemaker/core/workflow/step_outputs.py +65 -0
  334. sagemaker/core/workflow/utilities.py +514 -0
  335. sagemaker/lineage/__init__.py +33 -0
  336. sagemaker/lineage/action.py +28 -0
  337. sagemaker/lineage/artifact.py +28 -0
  338. sagemaker/lineage/context.py +28 -0
  339. sagemaker/lineage/lineage_trial_component.py +28 -0
  340. {sagemaker_core-1.0.62.dist-info → sagemaker_core-2.3.1.dist-info}/METADATA +28 -9
  341. sagemaker_core-2.3.1.dist-info/RECORD +351 -0
  342. sagemaker_core-2.3.1.dist-info/top_level.txt +1 -0
  343. sagemaker_core/_version.py +0 -3
  344. sagemaker_core/helper/session_helper.py +0 -769
  345. sagemaker_core/resources/__init__.py +0 -1
  346. sagemaker_core/shapes/__init__.py +0 -1
  347. sagemaker_core/tools/__init__.py +0 -1
  348. sagemaker_core-1.0.62.dist-info/RECORD +0 -35
  349. sagemaker_core-1.0.62.dist-info/top_level.txt +0 -1
  350. {sagemaker_core → sagemaker/core/helper}/__init__.py +0 -0
  351. {sagemaker_core/helper → sagemaker/core/jumpstart/factory}/__init__.py +0 -0
  352. {sagemaker_core/main → sagemaker/core/jumpstart/hub}/__init__.py +0 -0
  353. {sagemaker_core/main/code_injection → sagemaker/core/modules/local_core}/__init__.py +0 -0
  354. {sagemaker_core/main → sagemaker/core/utils}/code_injection/base.py +0 -0
  355. {sagemaker_core/main → sagemaker/core/utils}/code_injection/constants.py +0 -0
  356. {sagemaker_core/main → sagemaker/core/utils}/user_agent.py +0 -0
  357. {sagemaker_core-1.0.62.dist-info → sagemaker_core-2.3.1.dist-info}/WHEEL +0 -0
  358. {sagemaker_core-1.0.62.dist-info → sagemaker_core-2.3.1.dist-info}/licenses/LICENSE +0 -0
sagemaker/core/image_uri_config/huggingface-neuronx.json ADDED
@@ -0,0 +1,686 @@
1
+ {
2
+ "training": {
3
+ "processors": [
4
+ "trn"
5
+ ],
6
+ "version_aliases": {
7
+ "4.28": "4.28.1",
8
+ "4.34": "4.34.1",
9
+ "4.36": "4.36.2",
10
+ "4.43": "4.43.2",
11
+ "4.48": "4.48.1",
12
+ "4.51": "4.51.0",
13
+ "4.55": "4.55.4"
14
+ },
15
+ "versions": {
16
+ "4.28.1": {
17
+ "version_aliases": {
18
+ "pytorch1.13": "pytorch1.13.0"
19
+ },
20
+ "pytorch1.13.0": {
21
+ "py_versions": [
22
+ "py38"
23
+ ],
24
+ "repository": "huggingface-pytorch-training-neuronx",
25
+ "registries": {
26
+ "ap-east-2": "975050140332",
27
+ "ap-northeast-1": "763104351884",
28
+ "ap-south-1": "763104351884",
29
+ "ap-south-2": "772153158452",
30
+ "ap-southeast-1": "763104351884",
31
+ "ap-southeast-2": "763104351884",
32
+ "ap-southeast-4": "457447274322",
33
+ "ap-southeast-5": "550225433462",
34
+ "ap-southeast-7": "590183813437",
35
+ "cn-north-1": "727897471807",
36
+ "cn-northwest-1": "727897471807",
37
+ "eu-central-1": "763104351884",
38
+ "eu-central-2": "380420809688",
39
+ "eu-south-2": "503227376785",
40
+ "eu-west-1": "763104351884",
41
+ "eu-west-3": "763104351884",
42
+ "il-central-1": "780543022126",
43
+ "mx-central-1":"637423239942",
44
+ "sa-east-1": "763104351884",
45
+ "us-east-1": "763104351884",
46
+ "us-east-2": "763104351884",
47
+ "us-gov-east-1": "446045086412",
48
+ "us-gov-west-1": "442386744353",
49
+ "us-west-2": "763104351884",
50
+ "ca-west-1": "204538143572"
51
+ },
52
+ "container_version": {
53
+ "trn": "ubuntu20.04"
54
+ },
55
+ "sdk_versions": [
56
+ "sdk2.9.1"
57
+ ]
58
+ }
59
+ },
60
+ "4.34.1": {
61
+ "version_aliases": {
62
+ "pytorch1.13": "pytorch1.13.1"
63
+ },
64
+ "pytorch1.13.1": {
65
+ "py_versions": [
66
+ "py310"
67
+ ],
68
+ "repository": "huggingface-pytorch-training-neuronx",
69
+ "registries": {
70
+ "ap-east-2": "975050140332",
71
+ "ap-northeast-1": "763104351884",
72
+ "ap-south-1": "763104351884",
73
+ "ap-south-2": "772153158452",
74
+ "ap-southeast-1": "763104351884",
75
+ "ap-southeast-2": "763104351884",
76
+ "ap-southeast-4": "457447274322",
77
+ "ap-southeast-5": "550225433462",
78
+ "ap-southeast-7": "590183813437",
79
+ "cn-north-1": "727897471807",
80
+ "cn-northwest-1": "727897471807",
81
+ "eu-central-1": "763104351884",
82
+ "eu-central-2": "380420809688",
83
+ "eu-south-2": "503227376785",
84
+ "eu-west-1": "763104351884",
85
+ "eu-west-3": "763104351884",
86
+ "il-central-1": "780543022126",
87
+ "mx-central-1":"637423239942",
88
+ "sa-east-1": "763104351884",
89
+ "us-east-1": "763104351884",
90
+ "us-east-2": "763104351884",
91
+ "us-gov-east-1": "446045086412",
92
+ "us-gov-west-1": "442386744353",
93
+ "us-west-2": "763104351884",
94
+ "ca-west-1": "204538143572"
95
+ },
96
+ "container_version": {
97
+ "inf": "ubuntu20.04"
98
+ },
99
+ "sdk_versions": [
100
+ "sdk2.15.0"
101
+ ]
102
+ }
103
+ },
104
+ "4.36.2": {
105
+ "version_aliases": {
106
+ "pytorch1.13": "pytorch1.13.1"
107
+ },
108
+ "pytorch1.13.1": {
109
+ "py_versions": [
110
+ "py310"
111
+ ],
112
+ "repository": "huggingface-pytorch-training-neuronx",
113
+ "registries": {
114
+ "ap-east-2": "975050140332",
115
+ "ap-northeast-1": "763104351884",
116
+ "ap-south-1": "763104351884",
117
+ "ap-south-2": "772153158452",
118
+ "ap-southeast-1": "763104351884",
119
+ "ap-southeast-2": "763104351884",
120
+ "ap-southeast-4": "457447274322",
121
+ "ap-southeast-5": "550225433462",
122
+ "ap-southeast-7": "590183813437",
123
+ "cn-north-1": "727897471807",
124
+ "cn-northwest-1": "727897471807",
125
+ "eu-central-1": "763104351884",
126
+ "eu-central-2": "380420809688",
127
+ "eu-south-2": "503227376785",
128
+ "eu-west-1": "763104351884",
129
+ "eu-west-3": "763104351884",
130
+ "il-central-1": "780543022126",
131
+ "mx-central-1":"637423239942",
132
+ "sa-east-1": "763104351884",
133
+ "us-east-1": "763104351884",
134
+ "us-east-2": "763104351884",
135
+ "us-gov-east-1": "446045086412",
136
+ "us-gov-west-1": "442386744353",
137
+ "us-west-2": "763104351884",
138
+ "ca-west-1": "204538143572"
139
+ },
140
+ "container_version": {
141
+ "inf": "ubuntu20.04"
142
+ },
143
+ "sdk_versions": [
144
+ "sdk2.18.0"
145
+ ]
146
+ }
147
+ },
148
+ "4.43.2": {
149
+ "version_aliases": {
150
+ "pytorch2.1": "pytorch2.1.2"
151
+ },
152
+ "pytorch2.1.2": {
153
+ "py_versions": [
154
+ "py310"
155
+ ],
156
+ "repository": "huggingface-pytorch-training-neuronx",
157
+ "registries": {
158
+ "ap-northeast-1": "763104351884",
159
+ "ap-south-1": "763104351884",
160
+ "ap-south-2": "772153158452",
161
+ "ap-southeast-1": "763104351884",
162
+ "ap-southeast-2": "763104351884",
163
+ "ap-southeast-4": "457447274322",
164
+ "ap-southeast-5": "550225433462",
165
+ "ap-southeast-7": "590183813437",
166
+ "cn-north-1": "727897471807",
167
+ "cn-northwest-1": "727897471807",
168
+ "eu-central-1": "763104351884",
169
+ "eu-central-2": "380420809688",
170
+ "eu-south-2": "503227376785",
171
+ "eu-west-1": "763104351884",
172
+ "eu-west-3": "763104351884",
173
+ "il-central-1": "780543022126",
174
+ "mx-central-1":"637423239942",
175
+ "sa-east-1": "763104351884",
176
+ "us-east-1": "763104351884",
177
+ "us-east-2": "763104351884",
178
+ "us-gov-east-1": "446045086412",
179
+ "us-gov-west-1": "442386744353",
180
+ "us-west-2": "763104351884",
181
+ "ca-west-1": "204538143572"
182
+ },
183
+ "container_version": {
184
+ "inf": "ubuntu20.04"
185
+ },
186
+ "sdk_versions": [
187
+ "sdk2.20.0"
188
+ ]
189
+ }
190
+ },
191
+ "4.48.1": {
192
+ "version_aliases": {
193
+ "pytorch2.1": "pytorch2.1.2"
194
+ },
195
+ "pytorch2.1.2": {
196
+ "py_versions": [
197
+ "py310"
198
+ ],
199
+ "repository": "huggingface-pytorch-training-neuronx",
200
+ "registries": {
201
+ "ap-northeast-1": "763104351884",
202
+ "ap-south-1": "763104351884",
203
+ "ap-south-2": "772153158452",
204
+ "ap-southeast-1": "763104351884",
205
+ "ap-southeast-2": "763104351884",
206
+ "ap-southeast-4": "457447274322",
207
+ "ap-southeast-5": "550225433462",
208
+ "ap-southeast-7": "590183813437",
209
+ "cn-north-1": "727897471807",
210
+ "cn-northwest-1": "727897471807",
211
+ "eu-central-1": "763104351884",
212
+ "eu-central-2": "380420809688",
213
+ "eu-south-2": "503227376785",
214
+ "eu-west-1": "763104351884",
215
+ "eu-west-3": "763104351884",
216
+ "il-central-1": "780543022126",
217
+ "mx-central-1":"637423239942",
218
+ "sa-east-1": "763104351884",
219
+ "us-east-1": "763104351884",
220
+ "us-east-2": "763104351884",
221
+ "us-gov-east-1": "446045086412",
222
+ "us-gov-west-1": "442386744353",
223
+ "us-west-2": "763104351884",
224
+ "ca-west-1": "204538143572"
225
+ },
226
+ "container_version": {
227
+ "inf": "ubuntu20.04"
228
+ },
229
+ "sdk_versions": [
230
+ "sdk2.20.0"
231
+ ]
232
+ }
233
+ },
234
+ "4.51.0": {
235
+ "version_aliases": {
236
+ "pytorch2.7": "pytorch2.7.0"
237
+ },
238
+ "pytorch2.7.0": {
239
+ "py_versions": [
240
+ "py310"
241
+ ],
242
+ "repository": "huggingface-pytorch-training-neuronx",
243
+ "registries": {
244
+ "ap-northeast-1": "763104351884",
245
+ "ap-south-1": "763104351884",
246
+ "ap-south-2": "772153158452",
247
+ "ap-southeast-1": "763104351884",
248
+ "ap-southeast-2": "763104351884",
249
+ "ap-southeast-4": "457447274322",
250
+ "ap-southeast-5": "550225433462",
251
+ "ap-southeast-7": "590183813437",
252
+ "cn-north-1": "727897471807",
253
+ "cn-northwest-1": "727897471807",
254
+ "eu-central-1": "763104351884",
255
+ "eu-central-2": "380420809688",
256
+ "eu-south-2": "503227376785",
257
+ "eu-west-1": "763104351884",
258
+ "eu-west-3": "763104351884",
259
+ "il-central-1": "780543022126",
260
+ "mx-central-1":"637423239942",
261
+ "sa-east-1": "763104351884",
262
+ "us-east-1": "763104351884",
263
+ "us-east-2": "763104351884",
264
+ "us-gov-east-1": "446045086412",
265
+ "us-gov-west-1": "442386744353",
266
+ "us-west-2": "763104351884",
267
+ "ca-west-1": "204538143572"
268
+ },
269
+ "container_version": {
270
+ "inf": "ubuntu22.04"
271
+ },
272
+ "sdk_versions": [
273
+ "sdk2.24.1"
274
+ ]
275
+ }
276
+ },
277
+ "4.55.4": {
278
+ "version_aliases": {
279
+ "pytorch2.7": "pytorch2.7.0"
280
+ },
281
+ "pytorch2.7.0": {
282
+ "py_versions": [
283
+ "py310"
284
+ ],
285
+ "repository": "huggingface-pytorch-training-neuronx",
286
+ "registries": {
287
+ "ap-northeast-1": "763104351884",
288
+ "ap-south-1": "763104351884",
289
+ "ap-south-2": "772153158452",
290
+ "ap-southeast-1": "763104351884",
291
+ "ap-southeast-2": "763104351884",
292
+ "ap-southeast-4": "457447274322",
293
+ "ap-southeast-5": "550225433462",
294
+ "ap-southeast-7": "590183813437",
295
+ "cn-north-1": "727897471807",
296
+ "cn-northwest-1": "727897471807",
297
+ "eu-central-1": "763104351884",
298
+ "eu-central-2": "380420809688",
299
+ "eu-south-2": "503227376785",
300
+ "eu-west-1": "763104351884",
301
+ "eu-west-3": "763104351884",
302
+ "il-central-1": "780543022126",
303
+ "mx-central-1":"637423239942",
304
+ "sa-east-1": "763104351884",
305
+ "us-east-1": "763104351884",
306
+ "us-east-2": "763104351884",
307
+ "us-gov-east-1": "446045086412",
308
+ "us-gov-west-1": "442386744353",
309
+ "us-west-2": "763104351884",
310
+ "ca-west-1": "204538143572"
311
+ },
312
+ "container_version": {
313
+ "inf": "ubuntu22.04"
314
+ },
315
+ "sdk_versions": [
316
+ "sdk2.26.0"
317
+ ]
318
+ }
319
+ }
320
+ }
321
+ },
322
+ "inference": {
323
+ "processors": [
324
+ "inf"
325
+ ],
326
+ "version_aliases": {
327
+ "4.28": "4.28.1",
328
+ "4.34": "4.34.1",
329
+ "4.36": "4.36.2",
330
+ "4.43": "4.43.2",
331
+ "4.51": "4.51.3",
332
+ "4.55": "4.55.4"
333
+ },
334
+ "versions": {
335
+ "4.28.1": {
336
+ "version_aliases": {
337
+ "pytorch1.13": "pytorch1.13.0"
338
+ },
339
+ "pytorch1.13.0": {
340
+ "py_versions": [
341
+ "py38"
342
+ ],
343
+ "repository": "huggingface-pytorch-inference-neuronx",
344
+ "registries": {
345
+ "af-south-1": "626614931356",
346
+ "il-central-1": "780543022126",
347
+ "ap-east-1": "871362719292",
348
+ "ap-east-2": "975050140332",
349
+ "ap-northeast-1": "763104351884",
350
+ "ap-northeast-2": "763104351884",
351
+ "ap-northeast-3": "364406365360",
352
+ "ap-south-1": "763104351884",
353
+ "ap-south-2": "772153158452",
354
+ "ap-southeast-1": "763104351884",
355
+ "ap-southeast-2": "763104351884",
356
+ "ap-southeast-4": "457447274322",
357
+ "ap-southeast-5": "550225433462",
358
+ "ap-southeast-7": "590183813437",
359
+ "ca-central-1": "763104351884",
360
+ "cn-north-1": "727897471807",
361
+ "cn-northwest-1": "727897471807",
362
+ "eu-central-1": "763104351884",
363
+ "eu-central-2": "380420809688",
364
+ "eu-north-1": "763104351884",
365
+ "eu-west-1": "763104351884",
366
+ "eu-west-2": "763104351884",
367
+ "eu-west-3": "763104351884",
368
+ "eu-south-1": "692866216735",
369
+ "eu-south-2": "503227376785",
370
+ "me-south-1": "217643126080",
371
+ "mx-central-1":"637423239942",
372
+ "sa-east-1": "763104351884",
373
+ "us-east-1": "763104351884",
374
+ "us-east-2": "763104351884",
375
+ "us-gov-east-1": "446045086412",
376
+ "us-gov-west-1": "442386744353",
377
+ "us-iso-east-1": "886529160074",
378
+ "us-isob-east-1": "094389454867",
379
+ "us-west-1": "763104351884",
380
+ "us-west-2": "763104351884",
381
+ "ca-west-1": "204538143572"
382
+ },
383
+ "container_version": {
384
+ "inf": "ubuntu20.04"
385
+ },
386
+ "sdk_versions": [
387
+ "sdk2.9.1"
388
+ ]
389
+ }
390
+ },
391
+ "4.34.1": {
392
+ "version_aliases": {
393
+ "pytorch1.13": "pytorch1.13.1"
394
+ },
395
+ "pytorch1.13.1": {
396
+ "py_versions": [
397
+ "py310"
398
+ ],
399
+ "repository": "huggingface-pytorch-inference-neuronx",
400
+ "registries": {
401
+ "af-south-1": "626614931356",
402
+ "il-central-1": "780543022126",
403
+ "ap-east-1": "871362719292",
404
+ "ap-east-2": "975050140332",
405
+ "ap-northeast-1": "763104351884",
406
+ "ap-northeast-2": "763104351884",
407
+ "ap-northeast-3": "364406365360",
408
+ "ap-south-1": "763104351884",
409
+ "ap-south-2": "772153158452",
410
+ "ap-southeast-1": "763104351884",
411
+ "ap-southeast-2": "763104351884",
412
+ "ap-southeast-4": "457447274322",
413
+ "ap-southeast-5": "550225433462",
414
+ "ap-southeast-7": "590183813437",
415
+ "ca-central-1": "763104351884",
416
+ "cn-north-1": "727897471807",
417
+ "cn-northwest-1": "727897471807",
418
+ "eu-central-1": "763104351884",
419
+ "eu-central-2": "380420809688",
420
+ "eu-north-1": "763104351884",
421
+ "eu-west-1": "763104351884",
422
+ "eu-west-2": "763104351884",
423
+ "eu-west-3": "763104351884",
424
+ "eu-south-1": "692866216735",
425
+ "eu-south-2": "503227376785",
426
+ "me-south-1": "217643126080",
427
+ "mx-central-1":"637423239942",
428
+ "sa-east-1": "763104351884",
429
+ "us-east-1": "763104351884",
430
+ "us-east-2": "763104351884",
431
+ "us-gov-east-1": "446045086412",
432
+ "us-gov-west-1": "442386744353",
433
+ "us-iso-east-1": "886529160074",
434
+ "us-isob-east-1": "094389454867",
435
+ "us-west-1": "763104351884",
436
+ "us-west-2": "763104351884",
437
+ "ca-west-1": "204538143572"
438
+ },
439
+ "container_version": {
440
+ "inf": "ubuntu20.04"
441
+ },
442
+ "sdk_versions": [
443
+ "sdk2.15.0"
444
+ ]
445
+ }
446
+ },
447
+ "4.36.2": {
448
+ "version_aliases": {
449
+ "pytorch1.13": "pytorch1.13.1",
450
+ "pytorch2.1": "pytorch2.1.2"
451
+ },
452
+ "pytorch1.13.1": {
453
+ "py_versions": [
454
+ "py310"
455
+ ],
456
+ "repository": "huggingface-pytorch-inference-neuronx",
457
+ "registries": {
458
+ "af-south-1": "626614931356",
459
+ "il-central-1": "780543022126",
460
+ "ap-east-1": "871362719292",
461
+ "ap-east-2": "975050140332",
462
+ "ap-northeast-1": "763104351884",
463
+ "ap-northeast-2": "763104351884",
464
+ "ap-northeast-3": "364406365360",
465
+ "ap-south-1": "763104351884",
466
+ "ap-south-2": "772153158452",
467
+ "ap-southeast-1": "763104351884",
468
+ "ap-southeast-2": "763104351884",
469
+ "ap-southeast-4": "457447274322",
470
+ "ap-southeast-5": "550225433462",
471
+ "ap-southeast-7": "590183813437",
472
+ "ca-central-1": "763104351884",
473
+ "cn-north-1": "727897471807",
474
+ "cn-northwest-1": "727897471807",
475
+ "eu-central-1": "763104351884",
476
+ "eu-central-2": "380420809688",
477
+ "eu-north-1": "763104351884",
478
+ "eu-west-1": "763104351884",
479
+ "eu-west-2": "763104351884",
480
+ "eu-west-3": "763104351884",
481
+ "eu-south-1": "692866216735",
482
+ "eu-south-2": "503227376785",
483
+ "me-south-1": "217643126080",
484
+ "mx-central-1":"637423239942",
485
+ "sa-east-1": "763104351884",
486
+ "us-east-1": "763104351884",
487
+ "us-east-2": "763104351884",
488
+ "us-gov-east-1": "446045086412",
489
+ "us-gov-west-1": "442386744353",
490
+ "us-iso-east-1": "886529160074",
491
+ "us-isob-east-1": "094389454867",
492
+ "us-west-1": "763104351884",
493
+ "us-west-2": "763104351884",
494
+ "ca-west-1": "204538143572"
495
+ },
496
+ "container_version": {
497
+ "inf": "ubuntu20.04"
498
+ },
499
+ "sdk_versions": [
500
+ "sdk2.16.1"
501
+ ]
502
+ },
503
+ "pytorch2.1.2": {
504
+ "py_versions": [
505
+ "py310"
506
+ ],
507
+ "repository": "huggingface-pytorch-inference-neuronx",
508
+ "registries": {
509
+ "af-south-1": "626614931356",
510
+ "il-central-1": "780543022126",
511
+ "ap-east-1": "871362719292",
512
+ "ap-east-2": "975050140332",
513
+ "ap-northeast-1": "763104351884",
514
+ "ap-northeast-2": "763104351884",
515
+ "ap-northeast-3": "364406365360",
516
+ "ap-south-1": "763104351884",
517
+ "ap-south-2": "772153158452",
518
+ "ap-southeast-1": "763104351884",
519
+ "ap-southeast-2": "763104351884",
520
+ "ap-southeast-4": "457447274322",
521
+ "ap-southeast-5": "550225433462",
522
+ "ap-southeast-7": "590183813437",
523
+ "ca-central-1": "763104351884",
524
+ "cn-north-1": "727897471807",
525
+ "cn-northwest-1": "727897471807",
526
+ "eu-central-1": "763104351884",
527
+ "eu-central-2": "380420809688",
528
+ "eu-north-1": "763104351884",
529
+ "eu-west-1": "763104351884",
530
+ "eu-west-2": "763104351884",
531
+ "eu-west-3": "763104351884",
532
+ "eu-south-1": "692866216735",
533
+ "eu-south-2": "503227376785",
534
+ "me-south-1": "217643126080",
535
+ "mx-central-1":"637423239942",
536
+ "sa-east-1": "763104351884",
537
+ "us-east-1": "763104351884",
538
+ "us-east-2": "763104351884",
539
+ "us-gov-east-1": "446045086412",
540
+ "us-gov-west-1": "442386744353",
541
+ "us-iso-east-1": "886529160074",
542
+ "us-isob-east-1": "094389454867",
543
+ "us-west-1": "763104351884",
544
+ "us-west-2": "763104351884",
545
+ "ca-west-1": "204538143572"
546
+ },
547
+ "container_version": {
548
+ "inf": "ubuntu20.04"
549
+ },
550
+ "sdk_versions": [
551
+ "sdk2.18.0"
552
+ ]
553
+ }
554
+ },
555
+ "4.43.2": {
556
+ "version_aliases": {
557
+ "pytorch2.1": "pytorch2.1.2"
558
+ },
559
+ "pytorch2.1.2": {
560
+ "py_versions": [
561
+ "py310"
562
+ ],
563
+ "repository": "huggingface-pytorch-inference-neuronx",
564
+ "registries": {
565
+ "ap-northeast-1": "763104351884",
566
+ "ap-south-1": "763104351884",
567
+ "ap-south-2": "772153158452",
568
+ "ap-southeast-1": "763104351884",
569
+ "ap-southeast-2": "763104351884",
570
+ "ap-southeast-4": "457447274322",
571
+ "ap-southeast-5": "550225433462",
572
+ "ap-southeast-7": "590183813437",
573
+ "cn-north-1": "727897471807",
574
+ "cn-northwest-1": "727897471807",
575
+ "eu-central-1": "763104351884",
576
+ "eu-central-2": "380420809688",
577
+ "eu-south-2": "503227376785",
578
+ "eu-west-1": "763104351884",
579
+ "eu-west-3": "763104351884",
580
+ "il-central-1": "780543022126",
581
+ "mx-central-1":"637423239942",
582
+ "sa-east-1": "763104351884",
583
+ "us-east-1": "763104351884",
584
+ "us-east-2": "763104351884",
585
+ "us-gov-east-1": "446045086412",
586
+ "us-gov-west-1": "442386744353",
587
+ "us-west-2": "763104351884",
588
+ "ca-west-1": "204538143572"
589
+ },
590
+ "container_version": {
591
+ "inf": "ubuntu20.04"
592
+ },
593
+ "sdk_versions": [
594
+ "sdk2.20.0"
595
+ ]
596
+ }
597
+ },
598
+ "4.51.3": {
599
+ "version_aliases": {
600
+ "pytorch2.7": "pytorch2.7.1"
601
+ },
602
+ "pytorch2.7.1": {
603
+ "py_versions": [
604
+ "py310"
605
+ ],
606
+ "repository": "huggingface-pytorch-inference-neuronx",
607
+ "registries": {
608
+ "ap-northeast-1": "763104351884",
609
+ "ap-south-1": "763104351884",
610
+ "ap-south-2": "772153158452",
611
+ "ap-southeast-1": "763104351884",
612
+ "ap-southeast-2": "763104351884",
613
+ "ap-southeast-4": "457447274322",
614
+ "ap-southeast-5": "550225433462",
615
+ "ap-southeast-7": "590183813437",
616
+ "cn-north-1": "727897471807",
617
+ "cn-northwest-1": "727897471807",
618
+ "eu-central-1": "763104351884",
619
+ "eu-central-2": "380420809688",
620
+ "eu-south-2": "503227376785",
621
+ "eu-west-1": "763104351884",
622
+ "eu-west-3": "763104351884",
623
+ "il-central-1": "780543022126",
624
+ "mx-central-1":"637423239942",
625
+ "sa-east-1": "763104351884",
626
+ "us-east-1": "763104351884",
627
+ "us-east-2": "763104351884",
628
+ "us-gov-east-1": "446045086412",
629
+ "us-gov-west-1": "442386744353",
630
+ "us-west-2": "763104351884",
631
+ "ca-west-1": "204538143572"
632
+ },
633
+ "container_version": {
634
+ "inf": "ubuntu22.04"
635
+ },
636
+ "sdk_versions": [
637
+ "sdk2.24.1"
638
+ ]
639
+ }
640
+ },
641
+ "4.55.4": {
642
+ "version_aliases": {
643
+ "pytorch2.7": "pytorch2.7.1"
644
+ },
645
+ "pytorch2.7.1": {
646
+ "py_versions": [
647
+ "py310"
648
+ ],
649
+ "repository": "huggingface-pytorch-inference-neuronx",
650
+ "registries": {
651
+ "ap-northeast-1": "763104351884",
652
+ "ap-south-1": "763104351884",
653
+ "ap-south-2": "772153158452",
654
+ "ap-southeast-1": "763104351884",
655
+ "ap-southeast-2": "763104351884",
656
+ "ap-southeast-4": "457447274322",
657
+ "ap-southeast-5": "550225433462",
658
+ "ap-southeast-7": "590183813437",
659
+ "cn-north-1": "727897471807",
660
+ "cn-northwest-1": "727897471807",
661
+ "eu-central-1": "763104351884",
662
+ "eu-central-2": "380420809688",
663
+ "eu-south-2": "503227376785",
664
+ "eu-west-1": "763104351884",
665
+ "eu-west-3": "763104351884",
666
+ "il-central-1": "780543022126",
667
+ "mx-central-1":"637423239942",
668
+ "sa-east-1": "763104351884",
669
+ "us-east-1": "763104351884",
670
+ "us-east-2": "763104351884",
671
+ "us-gov-east-1": "446045086412",
672
+ "us-gov-west-1": "442386744353",
673
+ "us-west-2": "763104351884",
674
+ "ca-west-1": "204538143572"
675
+ },
676
+ "container_version": {
677
+ "inf": "ubuntu22.04"
678
+ },
679
+ "sdk_versions": [
680
+ "sdk2.26.0"
681
+ ]
682
+ }
683
+ }
684
+ }
685
+ }
686
+ }