How to do dask.array.apply_along_axis when the function has two parameters

Question

How can I apply_along_axis using dask.array when we have a function with two parameters. I am presenting this simple example but the dataset has 2 billion rows and would like to