using-xtrax - 构建、运行、分析并导出 xtrax JAX 管线
指导使用 xtrax 开发 JAX 管线，涵盖批处理、执行、训练、CLI 组合、配置、分析、稀疏化和提前编译导出。
标签
更新于: 2026-10-03能力
典型输入
典型输出
该技能可以做什么
- 验证 xtrax 兼容性
- 设计批处理计划
- 组合管线边界
- 验证计划拓扑
- 执行映射与扫描轴
- 配置 TOML 驱动运行
- 运行训练流程
- 使用 EDA 分析计划
- 推断签名与束
- 稀疏化模型
- 组合 CLI 动词
- 提前编译导出管线
输入
- JAX 管线代码
- 领域库代码
- TOML 配置
- 模型与输入数据
- CLI 参数
- 可选 xtrax 扩展依赖
输出
- 批处理计划与解释
- 拓扑验证结果
- 管线与训练运行状态
- 已保存的运行状态
- 基于 Zarr 的输出
- 原生、wasm32 或 SPIR-V 导出产物
- CLI 报告
要求
- 兼容 xtrax 0.4.0 的安装
- 兼容的 JAX 与 JAXLIB 版本
- 所需的 xtrax 依赖
- 用于可视化的 xtrax[eda]
- 用于 Zarr 接收器和摘要的 xtrax[io]
- 用于 CLI 动词的 xtrax[cli]
- 用于 IREE 导出的 xtrax[export]