fsdp-engine-expert

fsdp-engine-expert is an agent for Claude Code from areal-project/AReaL. It costs 37 tokens per session (2,816 once invoked), scanned A, original, Apache-2.0.

A usage and configuration expert for FSDPEngine, a distributed machine-learning training engine based on PyTorch FSDP2. It focuses on using the engine within the AReaL project.

In plain words
What is it for?
Use it for FSDPEngine setup, parallel-strategy choices, workflow integration, weight synchronization, memory optimization, and troubleshooting in AReaL.
Why use it?
It helps avoid configuration mistakes when training models across multiple devices or combining parallel training strategies.

Agent for Claude Code

Written for Claude Code: installed under .claude/. Also seen: model in frontmatter; reads .claude/ paths.

About the project

AReaL is an infrastructure system for training large language models with reinforcement learning, connecting model training to applications built around AI agents. Researchers and developers use it to train reasoning and agentic models through asynchronous workflows, and the catalogue add-ons support working with AReaL.

areal-project/AReaL · 5,729 stars · on GitHub · areal-ai.io

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 agents/areal-project/areal/fsdp-engine-expert
Clone the repo
git clone --depth 1 https://github.com/areal-project/AReaL

Made for: Claude Code.

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 fsdp-engine-expert

README.md
[![agentmods](https://agentmods.dev/badge/agents/areal-project/areal/fsdp-engine-expert.svg)](https://agentmods.dev/agents/areal-project/areal/fsdp-engine-expert)
Your own site
<a href="https://agentmods.dev/agents/areal-project/areal/fsdp-engine-expert"><img src="https://agentmods.dev/badge/agents/areal-project/areal/fsdp-engine-expert.svg" alt="Measured on agentmods" height="20"></a>
Per session 37 Only the description is in the session, so the agent can decide to use it. The body loads when it is invoked.
When invoked 2,816 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.1 $0.00037 $0.02816
Opus 5 $0.00018 $0.01408
Sonnet 5 $0.00007 $0.00563
Haiku 4.5 $0.00004 $0.00282

Measured 6d ago against content hash 18b5ac0e1b13, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-06, from the pricing page.

Security

Grade A, and why

fsdp-engine-expert 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 6d 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.

.claude/agents/fsdp-engine-expert.md · 297 lines

How it starts

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

FSDPEngine Usage Expert

You are an expert in FSDPEngine configuration and usage in AReaL. Focus on integration guidance, configuration patterns, and workflow usage.

When to Activate

Use for FSDPEngine usage guidance:

  • Configuration and parallel strategy setup
  • Workflow integration and weight synchronization
  • Performance optimization and memory management
  • Troubleshooting integration issues

Do NOT use for low-level implementation details or general distributed training theory.

Core Concepts

FSDPEngine is AReaL's general-purpose training engine based on PyTorch FSDP2. It provides distributed training for dense transformer models with integrated TP/DP/CP parallelism and memory optimization.

Key strengths:

  • FSDP2-based parameter sharding for memory efficiency
  • Support for TP (tensor), DP (data), and CP (context) parallelism
  • Algorithm-specific subclasses (PPO actor/critic, SFT, reward model)
  • CPU offloading and memory-efficient loading

Engine selection: Choose FSDPEngine for dense models, ArchonEngine for MoE models, MegatronEngine for pipeline-parallel ultra-deep models.

Configuration

Configuration Overview

FSDPEngine configuration combines TrainEngineConfig for training settings and ParallelStrategy for model parallelism.

Configuration Components:

  • TrainEngineConfig (areal/api/cli_args.py): Core training configuration with optimization parameters and engine-specific settings
  • ParallelStrategy (areal/api/alloc_mode.py): Defines parallel dimensions including tensor, data, and context parallelism sizes
  • FSDPEngineConfig (areal/api/cli_args.py): FSDP-specific settings including wrap policy, CPU offloading, and memory-efficient loading

Configuration Approach:

  1. Define model parallelism using ParallelStrategy with appropriate dimensions (TP, DP, CP)
  2. Configure training engine via TrainEngineConfig, including fsdp field for FSDPEngineConfig
  3. Set training-specific options like checkpoint format, weight update method, and data types

Read the full file on GitHub · 297 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. 6d ago First seen · 297 lines · 37 tokens per session scan A 18b5ac0e1b13

Subscribe to this mod's changes

fsdp-engine-expert is an agent published in the GitHub repository areal-project/AReaL (5,729 stars, last pushed today), licensed Apache-2.0. It adds 37 tokens to every session and 2,816 once invoked, about $0.0002 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-30.

Related

Other agents, from other repositories

llm_backends

AgentFly supports multiple LLM backends for text generation, each with their own configuration options. This module provides configuration classes for different backend types including vLLM, Verl, and OpenAI-compatible clients. Among them, Verl backend is designed for internal training usage. The Verl backend is the…

Agent-One-Lab/AgentFly · 0 tokens

fenic-feature-developer

Use this agent when the user requests help implementing new features, operations, expressions, or functionality for the Fenic DataFrame library. This includes:\n\n- Adding new DataFrame operations (e.g., 'add a pivot operation to DataFrame')\n- Creating new logical expressions (e.g., 'implement a regexextract…

typedef-ai/fenic · 0 tokens

Data Scientist

Develops ML models, engineers features, works with Snowpark notebooks and Cortex ML functions, and conducts statistical analysis. Invoke with $ds.

Snowflake-Labs/cocoplus · 31 tokens

mlops-engineer

Build ML pipelines, experiment tracking, and model registries. Implements MLflow, Kubeflow, and automated retraining. Handles data versioning and reproducibility. Use PROACTIVELY for ML infrastructure, experiment management, or pipeline automation.

echoVic/blade-code · 54 tokens

data-scientist

Data analysis expert for SQL queries, BigQuery operations, and data insights. Use proactively for data analysis tasks and queries.

echoVic/blade-code · 29 tokens

scg-mapper

Maps a set of connectors into the Source Capability Graph (SCG) via a deterministic state machine — introspect, parse structure, link entities, finalize. Indexes reachability only, never the data behind it.

bearlike/Assistant · 48 tokens