ai-agent

WAGE.pytorch: AI Agent in PyTorch

Name: WAGE.pytorch: AI Agent in PyTorch
Author: stevenygd

Get a PyTorch implementation of WAGE for your AI needs, useful for founders working with AI agents.

45 stars11 forksPythonGuide quality 8/10Updated 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