kernelize
Mode.TRAINING | Mode.TORCH_COMPILE
Also update docs and tests.
torch.compile
get_local_kernel