ChainRules.jl
ChainRules.jl copied to clipboard
Allow N-dimensional arrays in `sort` rules
I think the reason these weren't added initially is that sortperm says "The method accepting dims requires at least Julia 1.9."
Restricted tests to 1.9+. Otherwise it does not break anything.