replace convert as well

Author: Miha Zgubic

src/lib/number.jl

@@ -1,3 +1,10 @@
+function ChainRulesCore.rrule(
+    ::ZygoteRuleConfig, ::typeof(convert), T::Type{<:Real}, x::Real
+)
+    convert_pullback(Δ) = (NoTangent(), NoTangent(), Δ)
+    return convert(T, x), convert_pullback
+end
+
 function ChainRulesCore.rrule(
     ::ZygoteRuleConfig, ::typeof(Base.literal_pow), ::typeof(^), x::Number, ::Val{p}
 ) where {p}