v6e support (#179)

Author: Obliviour

src/xpk/core/system_characteristics.py

@@ -142,6 +142,79 @@ def get_system_characteristics(
         'h100-mega-80gb-8',
     ),
     # TPU system characteristics
+    # v6e
+    'v6e-1': SystemCharacteristics(
+        '1x1',
+        1,
+        'tpu-v6e-slice',
+        'ct6e-standard-1t',
+        1,
+        AcceleratorType['TPU'],
+        'v6e-1',
+    ),
+    'v6e-4': SystemCharacteristics(
+        '2x2',
+        1,
+        'tpu-v6e-slice',
+        'ct6e-standard-4t',
+        4,
+        AcceleratorType['TPU'],
+        'v6e-4',
+    ),
+    'v6e-8': SystemCharacteristics(
+        '2x4',
+        2,
+        'tpu-v6e-slice',
+        'ct6e-standard-4t',
+        4,
+        AcceleratorType['TPU'],
+        'v6e-8',
+    ),
+    'v6e-16': SystemCharacteristics(
+        '4x4',
+        4,
+        'tpu-v6e-slice',
+        'ct6e-standard-4t',
+        4,
+        AcceleratorType['TPU'],
+        'v6e-16',
+    ),
+    'v6e-32': SystemCharacteristics(
+        '4x8',
+        8,
+        'tpu-v6e-slice',
+        'ct6e-standard-4t',
+        4,
+        AcceleratorType['TPU'],
+        'v6e-32',
+    ),
+    'v6e-64': SystemCharacteristics(
+        '8x8',
+        16,
+        'tpu-v6e-slice',
+        'ct6e-standard-4t',
+        4,
+        AcceleratorType['TPU'],
+        'v6e-64',
+    ),
+    'v6e-128': SystemCharacteristics(
+        '8x16',
+        32,
+        'tpu-v6e-slice',
+        'ct6e-standard-4t',
+        4,
+        AcceleratorType['TPU'],
+        'v6e-128',
+    ),
+    'v6e-256': SystemCharacteristics(
+        '16x16',
+        64,
+        'tpu-v6e-slice',
+        'ct6e-standard-4t',
+        4,
+        AcceleratorType['TPU'],
+        'v6e-256',
+    ),
     # v5p
     'v5p-8': SystemCharacteristics(
         '2x2x1',