Proposed new feature or change: I wrote this rough test to compare the argmax implementation on boolean arrays to a for loop implementation over an array of booleans. For me the for loop is faster ...

In various conditions it would be helpful to be able to generate boolean masks that point to specific indices of an array. For example it can be combined with if_else compute functions to merge two ...