Implement slice specific functions in JAX version

I was thinking whether we should replicate the functions like slice_get_antenna_parameters() in the JAX version of the TemplateSynthesis class? Those functions can then be a shared interface between the two version and allow users to more easily check the parameters used in the synthesis process.