4 ms·
I'm surprised to see pytorch and Jax mentioned as alternatives but not numba : https://github.com/numba/numba https://github.com/numba/numba I've recently had
by curvilinear_m 2y ago
I'm surprised to see pytorch and Jax mentioned as alternatives but not numba : https://github.com/numba/numba https://github.com/numba/numba
I've recently had to implement a few kernels to lower the memory footprint and runtime of some pytorch function : it's been really nice because numba kernels have type hints support (as opposed to raw cupy kernels).
- killingtime74 2y agoNumba doesn't support GPU though
- catshamando 2y agoYes it does: https://numba.readthedocs.io/en/stable/cuda/kernels.html https://numba.readthedocs.io/en/stable/cuda/kernels.html
- killingtime74 2y agoah thanks, TIL
- the_svd_doctor 2y agoIt does (for NVIDIA at least)