feat: remove unnecessary unthunk

Author: simonmandlik

src/aggregations/aggregation_stack.jl

@@ -67,10 +67,6 @@ AggregationStack(fs::AbstractAggregation...) = AggregationStack(fs)
 
 Flux.@layer :ignore AggregationStack
 
-# function (a::AggregationStack)(x::Maybe{AbstractArray}, bags::AbstractBags, args...)
-#     reduce(vcat, (f(x, bags, args...) for f in a.fs))
-# end
-
 @generated function (a::AggregationStack{T})(x::Maybe{AbstractArray}, bags::AbstractBags, args...) where {T<:Tuple}
     l = T.parameters |> length
     chs = map(1:l) do i

src/aggregations/segmented_sum.jl

@@ -56,7 +56,6 @@ function segmented_sum_forw(x::AbstractMatrix, ψ::AbstractVector, bags::Abstrac
 end
 
 function segmented_sum_back(Δ, y, x, ψ, bags, w) 
-    Δ = unthunk(Δ)
     dx = zero(x)
     dψ = zero(ψ)
     dw = isnothing(w) ? ZeroTangent() : zero(w)