This version should clean up some remaining issues, and focus on cleaning up remaining issues with the JAX version, including:

• phase corrections
• mismatch in the amplitude
• memory optimisation
• including SlicedSynthesis objects in JAX

Additionally, other minor issues can also be resolved.