I have a list of a objects, each of which has a function to be applied on a slice of a jax.numpy.array.