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