EnzymeAD/Enzyme.jl

What it solves

Enzyme 的设计旨在解决为复杂软件高效地计算梯度（自动微分）的问题。它允许开发者计算使用高性能语言编写的函数导数，从而避免了为机器学习或科学计算任务而手动重写代码的需要。

How it works

它作为 LLVM 的一个插件，这意味着它在编译器的中间表示（IR）上执行自动微分。它通过对优化后的代码进行微分，可以实现达到或超过目前最先进的 AD 工具的性能。

Who it's for

此项目是为机器学习和高性能计算 (HPC) 领域的开发者和研究人员设计的，他们需要为现有的代码库提供快速、高效的梯度计算，特别是那些使用 Julia 或基于 LLVM 的语言的开发者。

Highlights

• 自有的高性能自动微分技术
• 用于对优化后的代码进行微分的 LLVM 插件架构
• 支持反向模式自动微分
• 针对 GPU 核心 (kernels) 和高性能计算环境进行了优化