Agent skill

hf-job-submit

Submit training jobs to Hugging Face cloud GPUs

Stars 163
Forks 31

Install this agent skill to your Project

npx add-skill https://github.com/majiayu000/claude-skill-registry/tree/main/skills/data/hf-job-submit

SKILL.md

HF Job Submit

Submit a Python training script to run on Hugging Face's cloud GPU infrastructure.

How to Use

bash
$CLAUDE_CONFIG_DIR/skills/hf-job-submit/scripts/hf-job-submit.sh <script_path> <hardware> <packages>

Arguments

Argument Description Example
script_path Path to Python training script /home/user/workspace/train.py
hardware GPU hardware type t4-small, a10g-large
packages Space-separated pip packages "trl transformers datasets"

Hardware Options

Type GPU VRAM Best For
t4-small T4 16GB Models < 1B
t4-medium T4 16GB Models 1-3B
a10g-small A10G 24GB Models 1-3B
a10g-large A10G 24GB Models 3-7B (LoRA)
a100-large A100 40GB Models 3-7B (LoRA)

Prerequisites

  • HF_TOKEN environment variable must be set
  • Hugging Face Pro or Team account with GPU Jobs access

Output

Returns the Job ID on success, which can be used with hf-job-status and hf-job-logs.

Example

bash
$CLAUDE_CONFIG_DIR/skills/hf-job-submit/scripts/hf-job-submit.sh \
  "/home/user/workspace/train.py" \
  "t4-small" \
  "trl transformers datasets peft accelerate"

Didn't find tool you were looking for?

Be as detailed as possible for better results