Numpy vectorization exercises for interviews (ML coding)

Hi,

Does anybody know of a good resource with numpy vectorization exercises? Something like leetcode but for numpy vectorization. The reason why I'm asking about this is because I had several companies asking these sort of questions in the coding interviews. You are given a math formula, and you have to rewrite it in numpy/pytorch/tensorflow vectorized form, so that it is computationally efficient. I need basically some exercises to practise broadcasting, converting operations to matmul, masking, and other tensor tricks.

I appreciate your help in advance.