This document is relevant for: Inf2
, Trn1
, Trn2
nki#
Decorators#
This decorator compiles a function to run on NeuronDevices. |
|
Benchmark a NKI kernel on a NeuronDevice by using |
|
Compile and run a NKI kernel on NeuronDevice without involving ML frameworks such as PyTorch and JAX. |
|
Simulate a nki kernel on CPU using a built-in simulator in Neuron Compiler. |
Types#
A tensor object represents a multidimensional, homogeneous array of fixed-size items |
This document is relevant for: Inf2
, Trn1
, Trn2