diff --git a/ext/EinExprsChainRulesCoreExt.jl b/ext/EinExprsChainRulesCoreExt.jl
index b5ae37f..e8b7882 100644
--- a/ext/EinExprsChainRulesCoreExt.jl
+++ b/ext/EinExprsChainRulesCoreExt.jl
@@ -3,6 +3,23 @@ module EinExprsChainRulesCoreExt
 using EinExprs
 using ChainRulesCore
 
-@non_differentiable einexpr(::Any...)
+for f in [
+    :head,
+    :args,
+    :nargs,
+    :inds,
+    :branches,
+    :leaves,
+    :suminds,
+    :parsuminds,
+    :einexpr,
+    :sumtraces,
+    :indshistogram,
+    :hyperinds,
+    :neighbours,
+    :select,
+]
+    @eval @non_differentiable EinExprs.$f(::Any...)
+end
 
 end
\ No newline at end of file