JAX feels like the new kid on the block. Every time that I see it is because it has made a project a loot faster (eg. WhisperJAX or the stable diffusion in JAX). Can anyone explain me why is it so fast is it because of parallelism?