Just-in-Time compilation for PyTorch

Hello!
Is there any Numba for NumPy counterpart for PyTorch? I’ve found this link, but couldn’t find any examples there.
Could you please share some simple examples?