linear-array | Returns a set of utility functions | Natural Language Processing library

I want to perform element-wise mathematical operations (e.g. sum, multiply..) on two Python lists containing numbers or multiple nested lists which may contain again numbers or lists and so on.

The shape of the two lists are equal when performing the operation. Furthermore, the result should be of the same shape as the two input lists.

A shape may differ in:

1. length,

2. width (i.e. number of nestings),

3. order (e.g. the lists starts with a number followed by a nested list, but it may also be that it starts with a nested list, followed by numbers).

The shape changes arbitrarily each time I want to perform a mathematical operation.

Is there a library or built-in functionality which can perform math operations on arbitrarily shaped lists?

In the past, I've implemented a customized piece of code for each different shape, similar to 1, 2, 3, 4, 5, 6, 7, 8, 9, but I was wondering if there is already a more general solution in the form of a library or reference code for this.

Example 1 (summation):