[Metal] Improved get_strided_index

[ivarflakstad]

Between 6.7% and 11.4% increased thrpt for strided metal kernels.

We should add either pre-compilation or load kernels with a dynamic library so we don't have to repeat the same impl for each kernel.