ZygoteRules -> ChainRules

Author: mcabbott

Project.toml

@@ -4,16 +4,15 @@ authors = ["Michael Abbott"]
 version = "0.0.8"
 
 [deps]
+ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 NamedDims = "356022a1-0364-5f58-8944-0da4b18d706f"
 OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
-Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
-ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
 
 [compat]
+ChainRulesCore = "0.10.9"
 NamedDims = "0.2.16"
 OffsetArrays = "1"
-ZygoteRules = "0.2"
 julia = "1.3"
 
 [extras]

src/LazyStack.jl

@@ -339,19 +339,19 @@ function LazyStack.stack(s::Symbol, args...)
     ensure_named(data, name_last)
 end
 
-#===== Zygote =====#
+#===== Gradients =====#
 
-using ZygoteRules: @adjoint
+using ChainRulesCore: ChainRulesCore, rrule, NoTangent
 
-@adjoint function stack(vec::AbstractArray{<:AbstractArray{<:Any,IN}}) where {IN}
-    stack(vec), Δ -> ([view(Δ, ntuple(_->(:),IN)..., Tuple(I)...) for I in eachindex(vec)],)
+function ChainRulesCore.rrule(::typeof(stack), vec::AbstractArray{<:AbstractArray{<:Any,IN}}) where {IN}
+    stack(vec), Δ -> (NoTangent(), [view(Δ, ntuple(_->(:),IN)..., Tuple(I)...) for I in eachindex(vec)],)
 end
 
-@adjoint function stack(tup::Tuple{Vararg{<:AbstractArray{<:Any,IN}}}) where {IN}
-    stack(tup), Δ -> (ntuple(i -> view(Δ, ntuple(_->(:),IN)..., i), length(tup)),)
+function ChainRulesCore.rrule(::typeof(stack), tup::Tuple{Vararg{<:AbstractArray{<:Any,IN}}}) where {IN}
+    stack(tup), Δ -> (NoTangent(), ntuple(i -> view(Δ, ntuple(_->(:),IN)..., i), length(tup)),)
 end
 
-@adjoint function stack(gen::Base.Generator)
+function ChainRulesCore.rrule(::typeof(stack), gen::Base.Generator)
     stack(gen), Δ -> error("not yet!")
 end