4 ms·
Do jax/tensorflow/pytorch work with numba? I.e. can you pass one of their arrays through a numba function and have it (a) not crash (b) support backprop?
by krastanov 4y ago
Do jax/tensorflow/pytorch work with numba? I.e. can you pass one of their arrays through a numba function and have it (a) not crash (b) support backprop?
- p1esk 4y agoSeems like pytorch does: https://stackoverflow.com/questions/63169760/how-can-i-use-numba-for-pytorch-tensors https://stackoverflow.com/questions/63169760/how-can-i-use-n...
- krastanov 4y agoIt seems it would not crash if you use some interface function, but it does not seem to support autodiff.
- Siira 4y agoJax already has its own tracing compiler which produces optimized code.