If you follow ML news and is trying to learn how things work you may have seem a framework called JAX The following question is: What is JAX? Before you go ask this to Google or ChatGPT, I suggest a better explanation here: