If the jax function doesn't use one of the arguments (common in gradient calculations), then the HLO parameter count and function parameters will mismatch.

We need a way to get jax to tell us which input argument maps to which HLO parameter.