Does numba works in pytorch?

Hi, I found numba libarary helps run code faster than usual python code. I am wondering, to make the training time faster, does Numba works for pytorch? If not, do you have any alternative to make it faster in terms of speed. Thanks