{"slug":"athe-kunal/agent-shaper","name":"PyTorch-FX Shaper","description":"The MCP server provides shape of tensors to convert PyTorch code to einsum and einops","category":"productivity","tags":[],"official":false,"stars":1,"transport":null,"install":[{"cmd":"pip install -e","imports":[]}],"tools":[{"name":"get_annotated_sources","description":"Run a setup script, trace the model, return shape-annotated source for the requested files"},{"name":"get_fx_shapes","description":"Return raw FX shape data for modules and functions"},{"name":"validate_rewrite","description":"Check a rewritten nn.Module class produces identical outputs (atol=1e-5)"},{"name":"validate_rewrite_function","description":"Check a rewritten standalone function produces identical outputs"},{"name":"validate_file_rewrite","description":"Validate all classes in a rewritten file in one call"},{"name":"save_fixtures","description":"Capture and persist forward-pass fixtures for later testing"},{"name":"generate_test_files","description":"Generate pytest files that assert output identity against saved fixtures"}],"env_vars":["OPENAI_API_KEY","OPENAI_BASE_URL"],"auth_type":"none","github":"https://github.com/Athe-kunal/Agent-Shaper","homepage":"","server_url":"","status":"active","source":"mcpservers.org","updated_at":"Thu May 28"}