infra-skills
GPU kernels, training memory estimates, and technical diagrams
All Skills in This Repository (5)
Pure Emerald Level Indicatorsmaterial-you-slides
Generate self-contained 1280x720 HTML slide decks with Material You styling.
tilelang-developer
Develop, optimize, and debug TileLang GPU kernels for AI workloads.
megatron-memory-estimator
Estimate GPU memory usage for Megatron dense and MoE models from HuggingFace configs.
slime-user
Automate end-to-end RL training workflows for language models with SLIME.
tikz-flowchart
Automates creation of standardized TikZ flowcharts for LaTeX documents.
Frequently Asked Questions
FAQPage SchemaHow 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โopenclaw
Run a personal AI assistant across your devices and chat apps
superpowers
Gives coding agents a disciplined workflow from idea to merged code
react
AI agent skills for building, testing, and porting React core