Trợ lý PyTorch Lightning
Sắp xếp code PyTorch thành LightningModule và cấu hình huấn luyện đa GPU, logging, checkpoint một cách chuẩn mực.
Làm được gì
Kỹ năng giúp chuyển đổi code PyTorch rườm rà sang cấu trúc PyTorch Lightning: LightningModule (training/validation/test/predict step), LightningDataModule (pipeline dữ liệu) và Trainer (đa GPU/TPU, mixed precision, gradient accumulation). Ngoài ra còn hướng dẫn callback (ModelCheckpoint, EarlyStopping, LearningRateMonitor), tích hợp logger W&B/TensorBoard/MLflow và chọn chiến lược phân tán DDP, FSDP hay DeepSpeed. Kho lưu trữ kèm sẵn 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
- Kỹ sư/nhà nghiên cứu deep learning có script huấn luyện dài và rối
- Người muốn mở rộng thí nghiệm từ 1 GPU lên nhiều GPU hoặc nhiều node
- Nhóm ML cần chuẩn hóa logging và khả năng tái lập thí nghiệm
- Người muốn refactor code prototype thành cấu trúc dự án chuyên nghiệp
Ví dụ sử dụng
- "Chuyển vòng lặp huấn luyện trong train.py sang LightningModule" → nhận module đã chia rõ các bước và
configure_optimizers() - "Model 7B tham số, 4 GPU thì nên dùng chiến lược nào?" → gợi ý cấu hình Trainer với FSDP và cách lưu checkpoint
- "Thêm WandbLogger, EarlyStopping và lưu model tốt nhất" → sinh code Trainer đầy đủ logger và callback
· · · Hướng dẫn cài đặt · · ·
Cài vào ứng dụng Claude (không cần terminal)
- Tải tệp ZIP bằng nút bên dưới.
- Trong Claude, mở Settings → Capabilities và bật 'Code execution and file creation'. (chỉ một lần)
- Vào Customize → Skills → + → 'Upload a skill' rồi tải tệp ZIP lên.
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.
- Mở terminal.
- Tải kho mã:
git clone https://github.com/K-Dense-AI/scientific-agent-skills.git - Tạo thư mục skill:
mkdir -p ~/.claude/skills - Sao chép kỹ năng:
cp -r scientific-agent-skills/skills/pytorch-lightning ~/.claude/skills/ - Cài thư viện:
pip install lightning(nếu dùng GPU hãy cài PyTorch bản CUDA trước) - Khởi động lại Claude Code và yêu cầu "viết code huấn luyện bằng PyTorch Lightning" để kích hoạt kỹ năng.