I document small JAX correctness investigations in JAX reliability notes: keyword forwarding in reusable autodiff and means over uneven replica groups. The material includes runnable CPU examples ...