nki.jit# nki.jit(func=None, mode='auto', **kwargs)[source]# This decorator compiles a top-level NKI function to run on NeuronDevices.