Unitful.Quantitys are not compatible with most array-based math rules in ChainRules.jl right now. Relevant issues discussing why this is are here and here.

We could solve this problem by duplicating the array rules from ChainRules.jl for properly typed Quanititys. A better solution would be to solve the aforementioned issues upstream in such a way that we can adopt the rules automatically