展开程序结构以检查编译过程

变更内容

本次小版本发布主要包含错误修复和改进.

  • 修复了 torch.compile 无法正确捕获 workflow.step 的问题。

  • 修复了 use_state 导致图中断(graph break)的问题。

  • 修复了一些模型缓冲区(buffer)使用不当的问题。

  • 修复了 monitor.plot 功能未按预期工作的问题。

  • 引入了一个新的包装器 evox.compile,以绕过 torch.compile 和 torch.vmap 的某些限制。

  • 解决了多个与 BatchedTensor 相关的问题:修复了对带有 EvalMonitor 的工作流应用 vmap 时可能导致 BatchedTensor 泄漏的错误。

  • 修复了导致 HPOProblem 无法与 BraxProblem 协同工作的问题。

  • 增强了 RVEA 和 CSO 的实现,以获得更好的性能和可靠性。

  • 增强了 BraxProblem 的实现。

  • 各种小的修复和改进。

相关资源

← 返回发布列表返回顶部 ↑