2 ms·
If you're happy with NumPy's API, then surely JAX is exactly what you're looking for.
by JBits 5mo ago
If you're happy with NumPy's API, then surely JAX is exactly what you're looking for.
- rao-v 5mo agoJAX can’t do what Numba can do for example. I just want one way to write simple math-y code like you normally would and automagically convert to run on one of the above approaches. That’s what compilers and high level languages are supposed to be for!