Implement `py_foundations_prob_3`. Libraries (NumPy, PyTorch, SciPy) are allowed, but Pure Python implementations earn bonus XP!
py_foundations_prob_3(*args, **kwargs)