yzlnewyzlnewCommunityยท5 Agent Skills Included

infra-skills

GPU kernels, training memory estimates, and technical diagrams

Writes high-performance GPU kernels with TileLang, estimates Megatron training memory for MoE and dense models, and configures SLIME reinforcement learning pipelines. Generates publication-quality TikZ architecture diagrams, flowcharts, and Material Design slides without manual layout work. Eliminates guesswork in GPU capacity planning and speeds up infrastructure documentation for engineering teams.
npx skills add yzlnew/infra-skills --all -g -y

All Skills in This Repository (5)

Pure Emerald Level Indicators

Frequently Asked Questions

FAQPage Schema
How to install infra-skills?โ–ผ

Run `npx skills add yzlnew/infra-skills --all -g -y` in your terminal to install all skills in this collection globally.

How to estimate GPU memory for Megatron training?โ–ผ

The megatron-memory-estimator skill calculates memory usage directly from HuggingFace configs and compares parallelism strategies like TP, PP, and EP to find a setup that fits your GPUs.

Can it write custom GPU kernels?โ–ผ

Yes. The tilelang-developer skill generates optimized TileLang kernels for GEMM, FlashAttention, and DeepSeek MLA on NVIDIA, AMD, and Ascend hardware.

How to create LLM architecture diagrams?โ–ผ

The hf-architecture-tikz skill turns any HuggingFace model config into a publication-quality TikZ diagram with parameter formulas and concrete numbers for every block.

Does infra-skills work with Claude Code?โ–ผ

Yes. All skills follow the standard SKILL.md format and activate automatically in Claude Code and compatible coding environments when relevant tasks are detected.

Related Repositories in Software Engineering

View All in Software Engineeringโ†’