diff --git a/ext/BijectorsEnzymeExt.jl b/ext/BijectorsEnzymeExt.jl index 1e8d8aa3..2c7fe4a6 100644 --- a/ext/BijectorsEnzymeExt.jl +++ b/ext/BijectorsEnzymeExt.jl @@ -3,6 +3,7 @@ module BijectorsEnzymeExt if isdefined(Base, :get_extension) using Enzyme: @import_frule, @import_rrule using Bijectors: find_alpha + using ChainRulesCore else using ..Enzyme: @import_frule, @import_rrule using ..Bijectors: find_alpha