diff --git a/task-gallery.json b/task-gallery.json index f8e99b8..689e610 100644 --- a/task-gallery.json +++ b/task-gallery.json @@ -10,6 +10,18 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -23,6 +35,18 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -36,6 +60,18 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -49,6 +85,18 @@ "framework": [ "unsloth" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -62,6 +110,18 @@ "framework": [ "unsloth" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -75,6 +135,18 @@ "framework": [ "unsloth" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -88,6 +160,18 @@ "framework": [ "diffusers" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -101,6 +185,18 @@ "framework": [ "sentence-transformers" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -114,6 +210,18 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -127,6 +235,18 @@ "framework": [ "nanochat" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -140,6 +260,18 @@ "framework": [ "lora" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -153,6 +285,18 @@ "framework": [ "trl" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -166,6 +310,14 @@ "framework": [ "mlx" ] + }, + "supportedAccelerators": { + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -180,6 +332,18 @@ "jax", "flax" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -191,6 +355,18 @@ "category": "training", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -204,6 +380,18 @@ "framework": [ "yolo" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -217,6 +405,32 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -230,6 +444,18 @@ "framework": [ "" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -243,6 +469,18 @@ "framework": [ "eleutherai" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -256,6 +494,18 @@ "framework": [ "deepeval" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -269,6 +519,18 @@ "framework": [ "" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -280,6 +542,32 @@ "category": "other", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -291,6 +579,32 @@ "category": "dataset-generation", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } } ] \ No newline at end of file diff --git a/tasks-gallery.json b/tasks-gallery.json index 3719831..93c8f33 100644 --- a/tasks-gallery.json +++ b/tasks-gallery.json @@ -10,6 +10,32 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -23,6 +49,18 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -36,6 +74,18 @@ "framework": [ "unsloth" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -49,6 +99,18 @@ "framework": [ "unsloth" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -62,6 +124,18 @@ "framework": [ "unsloth" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -75,6 +149,18 @@ "framework": [ "diffusers" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -88,6 +174,18 @@ "framework": [ "sentence-transformers" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -99,6 +197,32 @@ "category": "other", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -110,6 +234,32 @@ "category": "dataset-generation", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -123,6 +273,18 @@ "framework": [ "yolo" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -134,6 +296,18 @@ "category": "generation", "modality": "vision", "framework": [""] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -145,6 +319,18 @@ "category": "training", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -159,6 +345,18 @@ "jax", "flax" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -172,6 +370,18 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } } ] \ No newline at end of file diff --git a/tasks/1_trl_training_evals.json b/tasks/1_trl_training_evals.json index 45e60cc..ece9c1b 100644 --- a/tasks/1_trl_training_evals.json +++ b/tasks/1_trl_training_evals.json @@ -8,6 +8,18 @@ "category": "finetuning", "modality": "text", "framework": ["huggingface"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } } ] \ No newline at end of file diff --git a/tasks/2_ppo_training.json b/tasks/2_ppo_training.json index 93aeccc..998771c 100644 --- a/tasks/2_ppo_training.json +++ b/tasks/2_ppo_training.json @@ -8,6 +8,18 @@ "category": "rlhf", "modality": "text", "framework": ["huggingface"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } } ] \ No newline at end of file diff --git a/tasks/3_finetuning-llms.json b/tasks/3_finetuning-llms.json index 2cd8c59..073a669 100644 --- a/tasks/3_finetuning-llms.json +++ b/tasks/3_finetuning-llms.json @@ -8,6 +8,18 @@ "category": "finetuning", "modality": "text", "framework": ["huggingface"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -19,6 +31,18 @@ "category": "finetuning", "modality": "text", "framework": ["unsloth"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -30,6 +54,18 @@ "category": "training", "modality": "text", "framework": ["unsloth"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -43,6 +79,18 @@ "framework": [ "unsloth" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -56,6 +104,18 @@ "framework": [ "diffusers" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -69,6 +129,18 @@ "framework": [ "sentence-transformers" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -82,6 +154,18 @@ "framework": [ "huggingface" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -95,6 +179,18 @@ "framework": [ "nanochat" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -108,6 +204,18 @@ "framework": [ "lora" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -121,6 +229,18 @@ "framework": [ "trl" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -134,6 +254,14 @@ "framework": [ "mlx" ] + }, + "supportedAccelerators": { + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } } ] \ No newline at end of file diff --git a/tasks/4_mnist-jax-train.json b/tasks/4_mnist-jax-train.json index 2da1177..54ea3d8 100644 --- a/tasks/4_mnist-jax-train.json +++ b/tasks/4_mnist-jax-train.json @@ -10,6 +10,18 @@ "jax", "flax" ] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -21,6 +33,18 @@ "category": "training", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } } ] diff --git a/tasks/5_yolo.json b/tasks/5_yolo.json index 920805e..36caf4b 100644 --- a/tasks/5_yolo.json +++ b/tasks/5_yolo.json @@ -1,3 +1,4 @@ +[ { "title": "YOLO Object Detection Train Task Example", "description": "A sample YOLO object detection train task that runs a YOLO object detection train script on COCO8 dataset for 10 epochs", @@ -7,5 +8,18 @@ "category": "training", "modality": "vision", "framework": ["yolo"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } - } \ No newline at end of file + } +] \ No newline at end of file diff --git a/tasks/6_evals.json b/tasks/6_evals.json index 91c831f..dafa83c 100644 --- a/tasks/6_evals.json +++ b/tasks/6_evals.json @@ -8,6 +8,32 @@ "category": "eval", "modality": "text", "framework": ["huggingface"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -19,6 +45,18 @@ "category": "eval", "modality": "text", "framework": [""] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -30,6 +68,18 @@ "category": "eval", "modality": "text", "framework": ["eleutherai"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -41,6 +91,18 @@ "category": "eval", "modality": "text", "framework": ["deepeval"] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } } ] diff --git a/tasks/sample.json b/tasks/sample.json index 1fed4a1..d8b8dc5 100644 --- a/tasks/sample.json +++ b/tasks/sample.json @@ -8,6 +8,18 @@ "category": "generation", "modality": "vision", "framework": [""] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "accelerators": "RTX3090:1" + } + }, + "AMD": { + "resources": { + "accelerators": "7900XTX:1" + } + } } }, { @@ -19,6 +31,32 @@ "category": "other", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } }, { @@ -30,6 +68,32 @@ "category": "dataset-generation", "modality": "other", "framework": [] + }, + "supportedAccelerators": { + "NVIDIA": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AMD": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "AppleSilicon": { + "resources": { + "cpus": "2", + "memory": "4" + } + }, + "cpu": { + "resources": { + "cpus": "2", + "memory": "4" + } + } } } ]