https://www.somethingsblog.com/2024/10/21/mastering-argmin-a-powerful-tool-for-array-indexing/