ai-agent

WAGE.pytorch: Fast PyTorch Implementation of the WAGE Model for AI Agents

Name: WAGE.pytorch: Fast PyTorch Implementation of the WAGE Model for AI Agents
Author: stevenygd

Provides a faithful PyTorch reproduction of WAGE, letting AI‑focused founders quickly integrate the model into their pipelines without rewriting code.

From the repo · “Reproduction of WAGE in PyTorch.”

45 stars11 forksPythonGuide quality 8/10 · reviewed Oct 11Updated 11/18/2018100% free · open source

What it does

WAGE.pytorch is a PyTorch implementation of the WAGE algorithm, allowing for the reproduction of its results in a PyTorch environment.

When to use it

•When you need to reproduce WAGE results in a PyTorch environment
•When you want to leverage PyTorch's GPU acceleration for WAGE computations
•When you need to integrate WAGE with other PyTorch models or tools

Ready-to-paste prompt

python main.py --config config.py --gpu 0 --batch_size 32

Heads up: Make sure you have PyTorch installed and configured properly, as WAGE.pytorch relies on it for GPU acceleration and tensor computations

Saves to your device

Use with Claude

New

Skip the builder — one click puts this in Claude, Cursor, Antigravity and more.

✅ Light setup

Installs with a command or two; your AI agent can do it for you.

Try it instantly — no install

Claude Code

mkdir -p ~/.claude/skills/wage-pytorch && curl -fsSL https://workflowstacks.com/api/skills/wage-pytorch/claude-skill -o ~/.claude/skills/wage-pytorch/SKILL.md

Open in another AI app

Opens the app with this repo with the prompt ready to go — no copy-paste needed.

Connect the whole catalog (MCP)

claude mcp add --transport http workflowstacks https://workflowstacks.com/api/mcp

Adds a WorkflowStacks connector to Claude Code: search and load any skill here by chatting.

Use with Claude

New

Skip the builder — one click puts this in Claude, Cursor, Antigravity and more.

✅ Light setup

Installs with a command or two; your AI agent can do it for you.

Try it instantly — no install

Claude Code

mkdir -p ~/.claude/skills/wage-pytorch && curl -fsSL https://workflowstacks.com/api/skills/wage-pytorch/claude-skill -o ~/.claude/skills/wage-pytorch/SKILL.md

Open in another AI app

Opens the app with this repo with the prompt ready to go — no copy-paste needed.

Connect the whole catalog (MCP)

claude mcp add --transport http workflowstacks https://workflowstacks.com/api/mcp

Adds a WorkflowStacks connector to Claude Code: search and load any skill here by chatting.

Quick Actions

Details

Creator

stevenygd

Language

Python