Nice and concise example. Would love a mini-writeup / benchmarks for this (including the numba version).