Claude Skill MartKhám phá skillCâu dùng ngayHọc qua videoDùng terminalSkill là gì?
 
← Về danh sách

Trợ lý PyTorch Lightning

Skill giúp tổ chức code PyTorch thành LightningModule, Trainer, DataModule và thiết lập huấn luyện đa GPU, logging.

Lập trìnhTrung cấp★ 33,030⑂ 3,248Điểm AI 9/10Cập nhật lần cuối: 9 thg 8, 2026

Skill này làm gì

  • Chuyển code PyTorch rời rạc thành LightningModule với 6 phần rõ ràng (init, training/validation/test step, predict, optimizer).
  • Cấu hình Trainer: đa GPU/TPU, mixed precision, gradient accumulation & clipping, checkpoint, early stopping.
  • Đóng gói pipeline dữ liệu bằng LightningDataModule và kết nối logger W&B, TensorBoard, MLflow, CSV.
  • Tư vấn chiến lược phân tán theo kích thước model: DDP (<500M), FSDP (500M+), DeepSpeed.
  • Kèm template script và tài liệu tham chiếu chi tiết cho từng thành phần.

Phù hợp với ai

  • ML engineer muốn giảm boilerplate trong vòng lặp huấn luyện PyTorch.
  • Nhà nghiên cứu cần mở rộng thí nghiệm từ 1 GPU lên nhiều GPU/node.
  • Nhóm cần chuẩn hóa theo dõi thí nghiệm và quản lý checkpoint.

Ví dụ sử dụng

  1. "Chuyển vòng lặp epoch trong train.py sang LightningModule" → code được tách thành các step chuẩn.
  2. "Huấn luyện model 1.3B tham số trên 4 GPU thế nào?" → gợi ý FSDP, tham số Trainer và precision.
  3. "Lưu checkpoint tốt nhất theo val_loss, dừng nếu 5 epoch không cải thiện" → code ModelCheckpoint + EarlyStopping.

· · · Hướng dẫn cài đặt · · ·

Dùng thử ngay, không cần cài

Dán đoạn dưới vào Claude là dùng được skill này mà không cần cài đặt.

Hãy đọc hướng dẫn trong tệp này và làm theo để giúp mình:
https://raw.githubusercontent.com/K-Dense-AI/scientific-agent-skills/HEAD/skills/pytorch-lightning/SKILL.md

Mình muốn: (viết việc bạn cần ở đây)

Nếu Claude không mở được liên kết, hãy mở liên kết và copy nội dung vào rồi dán.

↓ Thấy hữu ích thì tải ZIP bên dưới và cài. Sau đó skill tự chạy, không phải dán lại mỗi lần.

Cài vào ứng dụng Claude (không cần terminal)
  1. Tải tệp ZIP bằng nút bên dưới.
  2. Trong Claude, mở Settings → Capabilities và bật 'Code execution and file creation'. (chỉ một lần)
  3. Vào Customize → Skills → + → 'Upload a skill' rồi tải tệp ZIP lên.
↓ Tải ZIP
Cài vào Claude Code

Để Claude làm — dán câu dưới đây vào Claude Code

Hãy cài skill mình tìm thấy trên Claude Skill Mart.
Copy thư mục skills/pytorch-lightning từ repo GitHub K-Dense-AI/scientific-agent-skills vào ~/.claude/skills/pytorch-lightning/ của mình.
Sau khi cài xong, cho mình biết skill này làm được gì trong một câu.

Cài bằng lệnh thủ công

git clone https://github.com/K-Dense-AI/scientific-agent-skills.git /tmp/sas && mkdir -p ~/.claude/skills && cp -r /tmp/sas/skills/pytorch-lightning ~/.claude/skills/

⚠ Đây là skill do người khác tạo. Hãy kiểm tra repo gốc trước khi cài.

  1. Mở terminal.
  2. Clone repo: git clone https://github.com/K-Dense-AI/scientific-agent-skills.git
  3. Tạo thư mục skill: mkdir -p ~/.claude/skills
  4. Copy skill: cp -r scientific-agent-skills/skills/pytorch-lightning ~/.claude/skills/
  5. Cài thư viện: pip install lightning (tùy chọn pip install wandb mlflow).
  6. Khởi động lại Claude Code và thử yêu cầu: "Tổ chức lại code huấn luyện bằng PyTorch Lightning".
Xem mã nguồn trên GitHub ↗Giấy phép: MIT