Why JAX? The NumPy You Know, But Faster
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...
kambale.dev9 min read