forward and reverse mode automatic differentiation primitives for Julia Base + StdLibs
The ChainRules package provides a variety of common utilities that can be used by downstream automatic differentiation (AD) tools to define and execute forward-, reverse-, and mixed-mode primitives.
The core logic of ChainRules is implemented in ChainRulesCore.jl. To add ChainRules support to your package, by defining new
frules, you only need to depend on the very light-weight package ChainRulesCore.jl. This repository contains ChainRules.jl, which is what people actually use directly. ChainRules reexports all the ChainRulesCore functionality, and has all the rules for the Julia standard library.
Here are some of the core features of the package: