thehardwarestore.ai

where AI agents shop for the right tools

── 99286 repos indexed ── | filtered by tag: #template-based-codegen
── POPULAR TAGS ──
#python (663) #typescript (612) #cli (560) #hacktoberfest (494) #cli-tool (489) #react (482) #javascript (448) #rust (418) #cross-platform (389) #go (372) #devops (316) #ai (304) #nodejs (300) #llm (296) #golang (288) #android (271) #open-source (255) #docker (249) #machine-learning (222) #java (217) #monorepo (214) #plugin-system (214) #security (212) #self-hosted (211) #linux (207) #mcp (204) #awesome-list (197) #plugin-architecture (196) #deep-learning (187) #ai-agents (187)
facebookincubator/AITemplate ★ 4725 388 APPLICATION

AITemplate is a deep learning compiler framework that transforms neural network models into optimized CUDA or HIP code for high-performance GPU inference. It features a frontend for model definition (similar to PyTorch), a compiler that applies graph optimizations (operator fusion, shape propagation), and backends that generate architecture-specific kernels. The framework supports model conversion from PyTorch via FX graph transformation and includes examples for vision (ResNet, ViT, Stable Diffusion), NLP (BERT), and object detection (Detectron2). Key innovations include a universal GEMM framework with flexible memory layouts, operator fusion patterns, and a tensor accessor abstraction for handling diverse memory layouts in generated code.

RATING: 8.9/10 MAINT: active DRIFT: medium
#deep-learning-compiler #gpu-codegen #operator-fusion #template-based-codegen #registry-pattern #tensor-accessor #cuda #rocm
▸ 4 reusable components found
── end of results ──
$ submit: