UED timing jitter calculation of arrival time jitter in UED The AD implementation requires jax and diffrax.