جوجل جاكس JAX مكتبة برمجية مفتوحة المصدر طوَّرتها شركة جوجل تتيح تحويل برامج بايثون وNumPy للعمل بكفاءة على وحدات المعالجة المركزية ووحدات المعالجة الرسومية، وتحافظ على هيكل NumPy وتتكامل مع أطر مثل TensorFlow وPyTorch، وتوفر ميزات مثل التمايز التلقائي (grad)، والتجميع JIT، والتوجيه الآلي vmap، وبرمجة SPMD عبر pmap.
