torchx-nightly 2025.3.11__py3-none-any.whl → 2025.3.13__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.

Potentially problematic release.

This version of torchx-nightly might be problematic. Click here for more details.

Files changed (7) hide show
  1. torchx/specs/named_resources_aws.py +44 -0
  2. {torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/METADATA +1 -1
  3. {torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/RECORD +7 -7
  4. {torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/LICENSE +0 -0
  5. {torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/WHEEL +0 -0
  6. {torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/entry_points.txt +0 -0
  7. {torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/top_level.txt +0 -0
torchx/specs/named_resources_aws.py CHANGED
@@ -368,6 +368,46 @@ def aws_trn1_32xlarge() -> Resource:
368
368
  )
369
369
 
370
370
 
371
+ def aws_inf2_xlarge() -> Resource:
372
+ return Resource(
373
+ cpu=4,
374
+ gpu=0,
375
+ memMB=16 * GiB,
376
+ capabilities={K8S_ITYPE: "inf2.xlarge"},
377
+ devices={NEURON_DEVICE: 1},
378
+ )
379
+
380
+
381
+ def aws_inf2_8xlarge() -> Resource:
382
+ return Resource(
383
+ cpu=32,
384
+ gpu=0,
385
+ memMB=128 * GiB,
386
+ capabilities={K8S_ITYPE: "inf2.8xlarge"},
387
+ devices={NEURON_DEVICE: 1},
388
+ )
389
+
390
+
391
+ def aws_inf2_24xlarge() -> Resource:
392
+ return Resource(
393
+ cpu=96,
394
+ gpu=0,
395
+ memMB=384 * GiB,
396
+ capabilities={K8S_ITYPE: "inf2.24xlarge"},
397
+ devices={NEURON_DEVICE: 6},
398
+ )
399
+
400
+
401
+ def aws_inf2_48xlarge() -> Resource:
402
+ return Resource(
403
+ cpu=192,
404
+ gpu=0,
405
+ memMB=768 * GiB,
406
+ capabilities={K8S_ITYPE: "inf2.48xlarge"},
407
+ devices={NEURON_DEVICE: 12},
408
+ )
409
+
410
+
371
411
  NAMED_RESOURCES: Mapping[str, Callable[[], Resource]] = {
372
412
  "aws_t3.medium": aws_t3_medium,
373
413
  "aws_m5.2xlarge": aws_m5_2xlarge,
@@ -405,4 +445,8 @@ NAMED_RESOURCES: Mapping[str, Callable[[], Resource]] = {
405
445
  "aws_g6e.48xlarge": aws_g6e_48xlarge,
406
446
  "aws_trn1.2xlarge": aws_trn1_2xlarge,
407
447
  "aws_trn1.32xlarge": aws_trn1_32xlarge,
448
+ "aws_inf2.xlarge": aws_inf2_xlarge,
449
+ "aws_inf2.8xlarge": aws_inf2_8xlarge,
450
+ "aws_inf2.24xlarge": aws_inf2_24xlarge,
451
+ "aws_inf2.48xlarge": aws_inf2_48xlarge,
408
452
  }
{torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: torchx-nightly
3
- Version: 2025.3.11
3
+ Version: 2025.3.13
4
4
  Summary: TorchX SDK and Components
5
5
  Home-page: https://github.com/pytorch/torchx
6
6
  Author: TorchX Devs
{torchx_nightly-2025.3.11.dist-info → torchx_nightly-2025.3.13.dist-info}/RECORD RENAMED
@@ -87,7 +87,7 @@ torchx/specs/api.py,sha256=ISdnfGH1R0-1Ah23UsPOZQT3WH63dIMj9CxA3QRoP0g,38046
87
87
  torchx/specs/builders.py,sha256=QDcQrnCO4bdSaiP0216XbCgTsnLutO_1_FW5jDiEIWI,9939
88
88
  torchx/specs/file_linter.py,sha256=IeiomB1BgHUlT-ZsvGxar3llY63NOupfLBrOrD_---A,11860
89
89
  torchx/specs/finder.py,sha256=MnwxG_UC4a-3X2wQ37ANEQR6D1TvriCLyuVYBh_-wuI,16249
90
- torchx/specs/named_resources_aws.py,sha256=a5vTnHYezsI32eOk5cePl4i1McJQ-hvQRG1IQXoO6VU,10415
90
+ torchx/specs/named_resources_aws.py,sha256=ISjHtifRJqB8u7PeAMiyLyO_S0WCaZiK-CFF3qe6JDU,11415
91
91
  torchx/specs/named_resources_generic.py,sha256=Sg4tAdqiiWDrDz2Lj_pnfsjzGIXKTou73wPseh6j55w,2646
92
92
  torchx/specs/test/components/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
93
93
  torchx/specs/test/components/a/__init__.py,sha256=kdxEgnI8QBSBiuTjaB4qDD7JX84hWowyPWU4B2Cqe9A,561
@@ -115,9 +115,9 @@ torchx/workspace/__init__.py,sha256=FqN8AN4VhR1C_SBY10MggQvNZmyanbbuPuE-JCjkyUY,
115
115
  torchx/workspace/api.py,sha256=PtDkGTC5lX03pRoYpuMz2KCmM1ZOycRP1UknqvNb97Y,6341
116
116
  torchx/workspace/dir_workspace.py,sha256=npNW_IjUZm_yS5r-8hrRkH46ndDd9a_eApT64m1S1T4,2268
117
117
  torchx/workspace/docker_workspace.py,sha256=PFu2KQNVC-0p2aKJ-W_BKA9ZOmXdCY2ABEkCExp3udQ,10269
118
- torchx_nightly-2025.3.11.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
119
- torchx_nightly-2025.3.11.dist-info/METADATA,sha256=F5XThGPtDgLgjHeRtSO74mCRo05GJk_dTWa04NheAmA,6169
120
- torchx_nightly-2025.3.11.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
121
- torchx_nightly-2025.3.11.dist-info/entry_points.txt,sha256=T328AMXeKI3JZnnxfkEew2ZcMN1oQDtkXjMz7lkV-P4,169
122
- torchx_nightly-2025.3.11.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
123
- torchx_nightly-2025.3.11.dist-info/RECORD,,
118
+ torchx_nightly-2025.3.13.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
119
+ torchx_nightly-2025.3.13.dist-info/METADATA,sha256=HkpmaUiV4SRP2UijDiMB-xBdL2q-sA7wzTOqyh62yzc,6169
120
+ torchx_nightly-2025.3.13.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
121
+ torchx_nightly-2025.3.13.dist-info/entry_points.txt,sha256=T328AMXeKI3JZnnxfkEew2ZcMN1oQDtkXjMz7lkV-P4,169
122
+ torchx_nightly-2025.3.13.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
123
+ torchx_nightly-2025.3.13.dist-info/RECORD,,