mit-ll-responsible-ai/hydra-zen

Create powerful Hydra applications without the yaml files and boilerplate code.

What it solves

hydra-zen 旨在简化可配置、可重复、且可扩展工作流的创建。它专门解决在 Hydra 项目中管理复杂配置文件（如手写的 YAML 文件）时的繁琐和耗时问题。

How it works

该库提供能够动态、自动生成基于 dataclass 的配置的函数。它还包括自定义的 config‑store API 和 task‑function 包装器，以减少项目中需要编写的 Hydra 相关样板代码量。

Who it’s for

它面向编写研究级或生产级代码的开发者和研究人员，尤其是使用 PyTorch Lightning 进行机器学习项目的用户。

Highlights

  • Eliminates YAML: 移除手写 YAML 配置文件的需求。
  • Configurable: 允许从单一接口（命令行或 Python 函数）管理代码的所有方面。
  • Repeatable: 自动将完整配置与结果一起保存，实现自我文档化的运行。
  • Scalable: 支持在本地或跨集群节点启动多个运行。