You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I need to evaluate the gradient of a function with my custom type/operations. For example if f(x) = x^3, I want the derivative obtained with the usual rules of calculus (i.e. df(x)=3*x^2), to be evaluated for typeof(x) = foo. Of course I have pre-defined the operations *(a::Number,b::foo) and ^(a::foo,n::Integer), etc...
I know that Zygote can easily do this, but it is about an order of magnitude slower than ReverseDiff for my problem and Real inputs, so I would rather prefer to use a compiled tape and ReverseDiff.
Thanks,
The text was updated successfully, but these errors were encountered:
Hi,
I need to evaluate the gradient of a function with my custom type/operations. For example if
f(x) = x^3
, I want the derivative obtained with the usual rules of calculus (i.e.df(x)=3*x^2)
, to be evaluated fortypeof(x) = foo
. Of course I have pre-defined the operations*(a::Number,b::foo)
and^(a::foo,n::Integer)
, etc...I know that Zygote can easily do this, but it is about an order of magnitude slower than ReverseDiff for my problem and
Real
inputs, so I would rather prefer to use a compiled tape and ReverseDiff.Thanks,
The text was updated successfully, but these errors were encountered: