Persistent Kernel + JIT-Injected Operators (CUDA) - View it on GitHub