To showcase the performance of JAX-based samplers for production PE, we should incorporate detector calibration inside the likelihood evaluation and sampling. A good starting place is the bilby source code, specifically: bilby/bilby/gw/detector/calibration.py.