用途:在改动变换或内核前先明确数学契约,记录形状、dtype、数值容差、随机性、可微性与所需后端;保留现有可用架构,使用当前公开接口而非兼容垫片。
主要能力:提供环境探测、项目静态扫描、编译探针、重编译探索与基准测试等脚本,覆盖追踪与 pytree、正确性调试与审查、计时与编译内存、放置与集合通信及多主机分片、自定义导数与扩展等参考主题。
使用方式:将 SKILL_DIR 解析为包含该文件的目录,脚本在其中而非目标项目 scripts/ 下,需在项目实际 Python 环境中运行,使用前查看各脚本帮助。示例:python "$SKILL_DIR/scripts/jax_env_report.py" --format json;python "$SKILL_DIR/scripts/jax_project_scan.py" /absolute/project --format json;python "$SKILL_DIR/scripts/jax_compile_probe.py" --help;python "$SKILL_DIR/scripts/jax_recompile_explorer.py" --help。基准测试需输入工厂:python "$SKILL_DIR/scripts/jax_benchmark_harness.py" --file /absolute/project/benchmark_case.py --function step --factory make_inputs --jit --donate-argnums 0 --repeat 20。
限制:环境探测会初始化后端并可能分配资源,报告只列环境变量名不列值,导入/后端/冒烟测试失败时返回非零,且非穷尽式密钥扫描,分享前需审查路径、设备与私有输出。静态扫描只给线索而非证明。导入、追踪、降级与基准测试会执行代码,需使用可信输入。捐赠会消耗输入缓冲区,独立计时须每次构造新输入。单设备运行无法验证多主机通信。资料未提供收费、价格与平台信息。