Skip to content

FluxML/ZygoteRules.jl

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
src
 
 
 
 
 
 
 
 
 
 
 
 

ZygoteRules.jl

This is a minimal (<100sloc) package which enables you to add custom gradients to Zygote, without depending on Zygote itself.

Usage:

foo(a, b) = a*b

using ZygoteRules

@adjoint foo(a, b) = a*b, c̄ -> (b'c̄, a'c̄)

See the Zygote docs for more details on how to write custom gradients.

About

No description, website, or topics provided.

Resources

License

Code of conduct

Stars

Watchers

Forks

Sponsor this project

 

Packages

No packages published

Languages