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: 支援在本機或叢集節點上同時啟動多個執行。