Improving DESC Surface Boozer Transform

[dpanici]

While we wait on progress on #369 , there are a few improvements that can be made when using the surface transform in DESC

• Adding utility function to perform the transform in the most efficient way possible (reusing matrices etc), similar to what was done in Add Ability to Write Boozmn.nc style output files #680
• Resolving Faster Zernike evaluation #720 to make zernike evaluations faster - this issue was actually reverted though...
• changing the utility functions used in calculating the transform at multiple surfaces (like in make_boozmn_output) to use jax.numpy, which can allow for more parts of the calculation to be performed on the GPU and potentially allow jitting more parts of the calculation (like the zernike_to_fourier function, for example) Vectorize boozer transform over multiple surfaces #1197

[dpanici]

#680 should resolve this maybe

[f0uriest]

With @unalmis recent updates to grid tags, I think we could add a requires_meshgrid thing to the boozer qtys and then vmap them over rho, allowing you to do multiple surfaces at once.