This will allow us to more easily vmap over the number of prony terms.
This will allow us to more easily vmap over the number of prony terms.