flyte-migrate-ml

flyte-migrate-ml is a skill for Claude Code, Codex from flyteorg/flyte-agent-plugins. It costs 95 tokens per session (5,638 once invoked), scanned A, original, Apache-2.0.

A guide for moving Flyte 1 machine-learning workloads to Flyte 2. Machine-learning workloads include model training, testing, hyperparameter search, GPU jobs, and batch predictions.

In plain words
What is it for?
Use it to migrate training and inference pipelines, hyperparameter optimization, GPU or deep-learning jobs, and batch inference to Flyte 2.
Why use it?
It addresses the differences involved in upgrading existing ML code and separates migration work from writing new Flyte 2 applications. It helps organize hardware and pipeline changes during the upgrade.

Skill for Claude CodeCodex

Part of the flyte plugin — 21 skills, 2 MCP servers shipped together

Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

agentmods
npx agentmods add skills/flyteorg/flyte-agent-plugins/flyte-migrate-ml
Any agent
npx skills add flyteorg/flyte-agent-plugins --skill flyte-migrate-ml
Clone the repo
git clone --depth 1 https://github.com/flyteorg/flyte-agent-plugins

Made for: Claude Code, Codex.

Or install flyte, the plugin that ships this one along with the rest of its 21 skills, 2 MCP servers.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for flyte-migrate-ml

README.md
[![agentmods](https://agentmods.dev/badge/skills/flyteorg/flyte-agent-plugins/flyte-migrate-ml.svg)](https://agentmods.dev/skills/flyteorg/flyte-agent-plugins/flyte-migrate-ml)
Your own site
<a href="https://agentmods.dev/skills/flyteorg/flyte-agent-plugins/flyte-migrate-ml"><img src="https://agentmods.dev/badge/skills/flyteorg/flyte-agent-plugins/flyte-migrate-ml.svg" alt="Measured on agentmods" height="20"></a>
Per session 95 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 5,638 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. Scan, not verified.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5 $0.00095 $0.05638
Opus 5 $0.00048 $0.02819
Sonnet 5 $0.00019 $0.01128
Haiku 4.5 $0.00010 $0.00564

Measured 3d ago against content hash bc62651f8b82, method: parsed. Prices are Anthropic first-party input rates as of 2026-08-30, from the pricing page.

Security

Grade A, and why

flyte-migrate-ml scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 3d ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

plugins/flyte/skills/flyte-migrate-ml/SKILL.md · 612 lines

How it starts

The opening of the file, as written. The whole thing — 612 lines — stays where its author put it; the contents beside it link to each section on GitHub.

Flyte 1 to Flyte 2 ML Migration Skill

Migrate existing Flyte 1 ML workloads — small-model training, hyperparameter optimization, deep learning on GPUs, and batch inference — to Flyte 2, then take advantage of patterns that were not possible in Flyte 1 (real-time serving, apps, sandboxed execution).

This skill is specifically about migrating existing v1 ML code. For greenfield authoring in Flyte 2, use the companion skills:

  • flyte-sdk-ml — writing new ML training / inference tasks in Flyte 2.
  • flyte-sdk-app — writing new apps and serving endpoints.
  • flyte-sdk-agent — writing new agents and sandboxed / code-mode workloads.

Grounding References

Resource URL
Migration guide (ML workloads) https://www.union.ai/docs/v2/flyte/user-guide/migration/flyte-2/ml-workloads/
Migration guide (New in Flyte 2) https://www.union.ai/docs/v2/flyte/user-guide/migration/flyte-2/new-in-flyte-2/
Official docs https://www.union.ai/docs/v2/flyte
Docs index (LLMs) https://www.union.ai/docs/v2/flyte/llms.txt
SDK API reference https://www.union.ai/docs/v2/union/api-reference/flyte-sdk/
Example code https://github.com/unionai/unionai-examples
Flyte MCP tools Available via flyte-mcp server

Migration Cheat Sheet

Flyte 1 Flyte 2
ImageSpec(name=..., packages=[...]) flyte.Image.from_debian_base().with_pip_packages(...)
@task(container_image=..., requests=..., cache=...) Set image, resources, cache once on flyte.TaskEnvironment, then @env.task
Resources(cpu=..., mem=...) flyte.Resources(cpu=..., memory=...) (note mem becomes memory)
Resources(gpu="1") + accelerator=T4 flyte.Resources(gpu="T4:1")
FlyteFile / FlyteFile(path=...) flyte.io.File / await File.from_local(...)
model_file.download() await model_file.download()
current_context().working_directory os.getcwd()
@workflow An orchestrating @env.task (plain async Python)
map_task(fn)(x=xs) await asyncio.gather(*[fn(x) for x in xs])
A "pick the best" task Plain Python after gather

Read the full file on GitHub · 612 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 3d ago First seen · 612 lines · 95 tokens per session scan A bc62651f8b82

Subscribe to this mod's changes

flyte-migrate-ml is a skill published in the GitHub repository flyteorg/flyte-agent-plugins (2 stars, last pushed 6d ago), licensed Apache-2.0. It adds 95 tokens to every session and 5,638 once invoked, about $0.0005 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-31.

Related

Other skills, from other repositories

bump-dependency

Bumps a Python package dependency across Home Assistant Core integrations, regenerates core requirement files, runs verification tests and prek lint, and prepares a pull request with proper release/compare links.

home-assistant/core · 42 tokens

mem0-vercel-ai-sdk

Mem0 provider for Vercel AI SDK (@mem0/vercel-ai-provider). TRIGGER when: user mentions "vercel ai sdk", "@mem0/vercel-ai-provider", "createMem0", "retrieveMemories", "addMemories", "getMemories", "searchMemories", "mem0 vercel", "AI SDK provider", "AI SDK memory", or is using generateText/streamText with mem0. Also…

mem0ai/mem0 · 146 tokens

schema-exploration

Lists tables, describes columns and data types, identifies foreign key relationships, and maps entity relationships in a database. Use when the user asks about database schema, table structure, column types, what tables exist, ERD, foreign keys, or how entities relate.

langchain-ai/deepagents · 57 tokens

deepagents-thread-inspector

Inspect and explain conversations in the local Deep Agents Code SQLite session store. Use as a fallback when LangSmith trace tooling is unavailable, for offline or untraced sessions, or when asked to identify or summarize a local dcode thread, inspect checkpoint metadata, list recent local threads, or parse…

langchain-ai/deepagents · 82 tokens

agent-framework-azure-ai-py

Build persistent agents on Azure AI Foundry using the Microsoft Agent Framework Python SDK.

sickn33/agentic-awesome-skills · 24 tokens

pause

Pause Mem0 memory capture on this machine. Use when the user wants to stop memories being recorded, for example for private work or experiments.

mem0ai/mem0 · 30 tokens