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 ...