We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5fe59ae commit a0df9b7Copy full SHA for a0df9b7
ext/BijectorsMooncakeExt.jl
@@ -1,7 +1,7 @@
1
module BijectorsMooncakeExt
2
3
using Mooncake:
4
- @is_primitive, MinimalCtx, Mooncake, CoDual, primal, tangent_type, @from_rrule
+ @is_primitive, MinimalCtx, Mooncake, CoDual, primal, tangent_type, @from_chainrules
5
using Bijectors: find_alpha, ChainRulesCore
6
7
@from_chainrules(MinimalCtx, Tuple{typeof(find_alpha),Float16,Float16,Float16})
0 commit comments