functional-conversion
CommunityTurn PyTorch modules into clean functional code.
Software Engineering#pytorch#convert#functional#code-transformation#dsl-generation#nn.module#model-rewrite
AuthorJust-it
Version1.0.0
Installs0
System Documentation
What problem does it solve?
Converts PyTorch nn.Module classes to functional API style, removing class wrappers and self parameters to produce a concise, testable functional implementation for DSL generation.
Core Features & Use Cases
- Functionalizes modules by replacing Module.forward calls with a module_fn that uses torch.nn.functional operations, preserving behavior.
- Generates a minimal, testable functional model including imports, a Model(nn.Module) wrapper, and helper input getters for quick validation.
- Enables DSL-based code generation, deployment, and reproducibility by providing a consistent functional interface across different ops via the provided references.
Quick Start
Convert a PyTorch nn.Module reference and its related references into a functional PyTorch implementation suitable for DSL pipelines.
Dependency Matrix
Required Modules
None requiredComponents
references
💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: functional-conversion Download link: https://github.com/Just-it/AscendOpGenAgent/archive/main.zip#functional-conversion Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.