Why JAX? The NumPy You Know, But Faster
Feb 2 · 9 min read · If you've been doing machine learning in Python for any length of time, you've written code like this: import numpy as np x = np.random.randn(1000, 1000) y = np.random.randn(1000, 1000) result = np.dot(x, y) NumPy is comfortable. It's the first thi...
Join discussion

