nki.jit#

nki.jit(func=None, mode='auto', **kwargs)[source]#

This decorator compiles a top-level NKI function to run on NeuronDevices.