Need help with ChainRules.jl?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

JuliaDiff
151 Stars 38 Forks Other 498 Commits 69 Opened issues

Description

forward and reverse mode automatic differentiation primitives for Julia Base + StdLibs

Services available

!
?

Need anything else?

Contributors list

ChainRules

CI Travis Codecov Coveralls PkgEval Code Style: Blue ColPrac: Contributor's Guide on Collaborative Practices for Community Packages

Docs:

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

rrule
s or
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:

  • Mixed-mode composability without being coupled to a specific AD implementation.
  • Extensible rules: package authors can add rules (and thus AD support) to the functions in their packages, without needing to make a PR to ChainRules.jl .
  • Control-inverted design: rule authors can fully specify derivatives in a concise manner that supports computational efficiencies, so we will only compute as much as the user requests.
  • Propagation semantics built-in, with default implementations that allow rule authors to easily opt-in to common optimizations (fusion, increment elision, memoization, etc.).

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.