I don’t fully understand the use case.
PyTorch modules are not single CUDA kernels, but are implemented in e.g. Python, call into the C++ backend, and finally dispatch to the corresponding CUDA kernels.
“Calling nn model directly” in a CUDA kernel would assume that “model” is a single kernel.
Could you explain your use case a bit more?