meta-pytorch/torchrec

test-gen

Generate tests for TorchRec source files with correct patterns (unit, distributed, hypothesis), proper BUCK targets, and test utilities. Use when asked to generate tests, add test coverage, or write tests for a module.

Langue du texte source : anglais

pr-review

Review TorchRec pull requests and diffs for distributed correctness, sharding safety, backward compatibility, and test coverage. Use when reviewing PRs, diffs, or when asked to review code changes.

Langue du texte source : anglais

sharding-stats

Investigate and explain TorchRec planner sharding statistics output, especially how HBM storage is computed per table and per rank. Use when the user asks about sharding stats, storage breakdown, or memory estimation.

Langue du texte source : anglais

docstring

Write docstrings for TorchRec functions and methods following PyTorch conventions. Use when writing or updating docstrings in TorchRec code.

Langue du texte source : anglais

skill-writer

Guide users through creating Agent Skills for Claude Code. Use when the user wants to create, write, author, or design a new Skill for TorchRec, or needs help with SKILL.md files.

Langue du texte source : anglais

create-spec

Interview user in-depth to create a detailed spec with strict implementation details and tradeoff analysis

Langue du texte source : anglais

techdebt

Find and remove tech debt (redundant/duplicated code), run linters, and ensure code quality in recent changes

Langue du texte source : anglais