This is a package containing the Julia bindings for Enzyme. This is very much a work in progress and bug reports/discussion is greatly appreciated!
Enzyme is a plugin that performs automatic differentiation (AD) of statically analyzable LLVM. It is highly-efficient and its ability perform AD on optimized code allows Enzyme to meet or exceed the performance of state-of-the-art AD tools.
Enzyme.jl can be installed in the usual way Julia packages are installed
] add Enzyme
Enzyme.jl can be used by calling autodiff
on a function to be differentiated as shown below:
using Enzyme
f1(x) = x*x
@test autodiff(f1, Active(1.0)) โ 1.0
More information on installing and using Enzyme can be found on our website: https://enzyme.mit.edu.
To get involved or if you have questions, please join our mailing list.
If using this code in an academic setting, please cite the following paper to appear in NeurIPS 2020
@incollection{enzymeNeurips,
title = {Instead of Rewriting Foreign Code for Machine Learning, Automatically Synthesize Fast Gradients},
author = {Moses, William S. and Churavy, Valentin},
booktitle = {Advances in Neural Information Processing Systems 33},
year = {2020},
note = {To appear in},
}