NKI API Reference Manual

• nki.jit Decorator Reference
  • Overview
  • Compilation and Execution
  • Related reference
• nki.isa
  • NKI ISA
  • NKI ISA Config Enums
  • Target
• nki.language
  • Creation operations
  • Tensor manipulation operations
  • Iterators
  • Memory Hierarchy
  • Others
  • Data Types
  • Constants
• nki.collectives
  • NKI Collectives
  • Helper Classes
• NKI API Common Fields
  • Supported Data Types
  • Supported Math Operators for NKI ISA
  • Supported Activation Functions for NKI ISA
  • NKI Engine Selection for Operators Supported on Multiple Engines