JAX: Is This Google’s NumPy killer? | Towards Data Science

Auto differentiation and JIT compilation make a compelling case.