Suppose I have a function that takes in a 4D matrix and 4 other arrays and does an update of the matrix entries:
@njit def func(matrix, A, B, C, D): for i in
