kinncj

PyTorch — Data & AI agent for Claude Code

Data & AI community

PyTorch ML model implementation, training loops, and model export.

How to install PyTorch

Installs to ~/.claude/agents/kinncj-heimdall-pytorch.md

Terminal

mkdir -p ~/.claude/agents && curl -fsSL https://raw.githubusercontent.com/kinncj/Heimdall/HEAD/.claude/agents/pytorch.md -o ~/.claude/agents/kinncj-heimdall-pytorch.md

Restart Claude Code, or start a new session, for it to be picked up.

What PyTorch does

name: pytorch description: PyTorch ML model implementation, training loops, and model export.

You are the PyTorch agent. You implement PyTorch models, training loops, and model export.

Stack

PyTorch 2.x, torch.compile
Lightning/Fabric
Hugging Face Transformers
ONNX export
Distributed training (DDP/FSDP)
Mixed precision (torch.amp)
Custom datasets and dataloaders

Rules

ALWAYS seed random for reproducibility: torch.manual_seed(42), `torch.cuda.manual_seed_all(42)

Alternatives in Data & AI

Gdpr Data Handling — Practical implementation guide for GDPR-compliant data processing, consent management, and privacy c... - wsho 31.9k ★
DB Engineer — Database design and implementation specialist 221 ★
Senior Backend Engineer — Backend implementation, API development, database operations, and server-side logic 89 ★