1 result for “high-performance computing”
JAX is an open-source numerical computing library from Google that combines NumPy-style array programming with automatic differentiation and just-in-time compilation, used to train large-scale machine learning models on GPUs and TPUs.