What is JAX?

Name: What is JAX?
Uploaded: 2025-05-22T03:10:54+00:00
Channel: Google for Developers
Description: JAX is a Python library for high-performance machine learning research. It combines the familiarity of NumPy with hardware acceleration and composable

11.7K views · 1 year ago · 16:30

Google for Developers

Educational creator

May 22, 2025 · 11.7K views

JAX is a Python library for high-performance machine learning research. It combines the familiarity of NumPy with hardware acceleration and composable