AGENT SKILL

pinn_residual_problem

Build deterministic PINN residual problem fixtures with boundary and PDE residual targets for NTK recovery experiments.

License 许可
Apache-2.0
最近更新
2026-09-03
Stars
★ 351
Forks
27

01 前言 Front Matter

name
pinn_residual_problem
description
Build deterministic PINN residual problem fixtures with boundary and PDE residual targets for NTK recovery experiments.

02 资源文件 3

03 SKILL.md 正文

PINN Residual Problem

Use this skill when a recovery task needs an explicit physics-informed neural network problem item rather than a generic regression dataset. It is appropriate for small Poisson-style probes, source-boundary checks, and module tests that need separated boundary and residual loss contracts.

Do not use this skill to claim a full paper benchmark. The default fixture is a declared synthetic proxy designed to exercise the paper mechanism.

Inputs

  • frequency: positive integer or float for u(x)=sin(frequency*pi*x).
  • residual_points: interior collocation points in [0, 1].
  • Optional boundary points, defaulting to [0, 1].

Outputs

  • A JSON-compatible problem dictionary with boundary_points, boundary_targets, residual_points, residual_targets, and operator.
  • Residual targets follow the Poisson second-derivative convention u_xx(x) = -(frequency*pi)^2 sin(frequency*pi*x).

Workflow

  1. Choose a deterministic set of boundary and residual points.
  2. Compute exact solution values and second-derivative forcing values.
  3. Validate that arrays are non-empty, finite, and aligned.
  4. Pass the problem dictionary to NTK diagnostics or proxy training.

Validation

Run:

python tests/test_poisson_problem.py

The tests verify endpoint boundary values, residual sign, finite values, and metadata needed by downstream recovery.

Limitations

The fixture is intentionally small and synthetic. It preserves the paper's residual-plus-boundary loss mechanism but does not reproduce full-width neural network training or all numerical experiments.

04 同仓库更多 Skills

VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-triton-opt

Make a single PassNet Triton kernel fast AND numerically faithful: performance model (when a replacement can win at all), block/grid/warp tuning, autotune policy, launch overhead, and per-op numeric recipes to pass the dtype baseline tolerances. Use when a pass matches and is correct but speedup below expected, or when correctness fails by small numeric margins.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-solve

END-TO-END playbook for solving one PassNet sample: analyze the computation graphs, decide the optimization strategy (what to fuse, what to replace, what to leave alone), drive the iteration loop, and maximize the sample score. This is the ENTRY skill — invoke it first for any PassNet optimization task; it tells you when to use passnet-pattern-fusion, passnet-triton-opt and passnet-feedback.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-skill

PassNet GPU kernel optimization via compiler passes. Design and implement Triton-based optimization passes, create pass files under ./pass_dir/, self-evaluate with pass_evaluator, and iterate to maximize GPU speedup.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-pattern-fusion

Author PassNet pass files: write patterns that actually MATCH the FX graph, pick fusion regions (which ops to absorb into one Triton kernel), and structure multi-pass submissions with the mandatory shared-dispatch architecture. Use when creating or fixing pattern()/replacement_args()/replacement_func() files, when a pass "failed to match", or when deciding how to fuse multiple kernels/ops.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-orchestrate

Round/batch planning and decision support for PassNet graph optimization. Use for multi-sample triage, worker allocation, eval-budget policy, and post-eval keep/revert decisions. For solving one concrete sample end to end, use passnet-solve as the entry skill; consult this skill only when you need broader planning or a specific decision gate.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-feedback

Fast, deterministic feedback for PassNet work: pre-flight pattern/match verification WITHOUT burning a GPU evaluation, per-node bottleneck analysis of a sample's graphs, and eval-log parsing into per-variant status + estimated score + failure classification. Use BEFORE every GPU evaluation (check_pattern), at the START of a sample (analyze_graph), and AFTER every evaluation (parse_eval_log).

★ 351