Idea: remove `numba` as a dependency

[scottstanie]

Meta-issue to track the removal of Numba.
There's only a few places we are using Numba instead of JAX for a compiled speedup. Having two accelerators seems excessive for the dependency list.

• Similarity
• interpolation
• _ps_filling.get_max_idxs
• ks SHP test