JAX is an open-source numerical computing library developed by Google. It is designed to combine the performance of machine learning frameworks like TensorFlow with the expressiveness of Python and NumPy.