How to include functions Zygote doesn't know about? about zygote.jl HOT 2 CLOSED

julia> using SLEEF, BenchmarkTools, ForwardDiff

julia> using Zygote: @grad, derivative
[ Info: Precompiling Zygote [e88e6eb3-aa80-5325-afca-941959d7151f]
[ Info: Precompiling IRTools [7869d1d1-7146-5819-86e3-90919afe41df]

julia> @grad SLEEF.sin(x) = (SLEEF.sin(x), Δ -> (Δ * SLEEF.cos(x),))

julia> foo(x) = sin(x)^2 + exp(sin(x)/4)/12.3
foo (generic function with 1 method)

julia> sfoo(x) = SLEEF.sin(x)^2 + exp(SLEEF.sin(x)/4)/12.3
sfoo (generic function with 1 method)

julia> @btime ForwardDiff.derivative(foo, 1.2)
  40.227 ns (0 allocations: 0 bytes)
0.6847607046969565

julia> @btime derivative(foo, 1.2)
  48.593 ns (1 allocation: 16 bytes)
0.6847607046969566

julia> @btime derivative(sfoo, 1.2)
  116.762 ns (1 allocation: 16 bytes)
0.6847607046969566

@grad function SLEEF.sin(x)
    sinx, cosx = SLEEF.sincos(x)
    sinx, cosx
    sinx, Δ -> (Δ * cosx, )
end