Re:PyTorch 1.0 is out
by mlthoughts2018 on Saturday December 08, 2018 @10:01AM
(#18635510)
I’ve had the opposite experience with numba in production. It works almost flawlessly, very easy to reason about the generated code and inspect annotations, easy to debug.
[ Parent
| on HN ]
Do you interact with numpy or other compiled numerical packages? this is where it usually breaks for me. The thing is that I use numpy, scipy, keras, tensorflow, etc. in literally every project, making numba not too useful
Check out github.com/google/jax, it’s NumPy on the GPU with automatic differentiation, JIT and autobatching.
Re:PyTorch 1.0 is out
by mlthoughts2018 on Saturday December 08, 2018 @04:05PM
(#18637532)
That’s very cool. Numba and Cython work extremely well with virtually no overhead or extra effort on my part, so jax doesn’t seem like it would buy me much for most of my work. But I can imagine a lot of projects where jax woukd be useful, and I plan to keep current on best practices for it.