4 ms·

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)

by snats 3y ago

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?

bigdict 3y ago
Because of XLA, a deep learning compiler that looks at the entire computational graph as a whole.