This document is relevant for: Trn2, Trn3

NKI API Reference Manual

• nki
  • nki.jit
  • nki.simulate
• nki.isa
  • NKI ISA
  • NKI ISA Config Enums
  • Target
  • Constants
• nki.language
  • Creation operations
  • Tensor operations
  • Math operations
  • Activation and Backpropagation functions
  • Normalization and Regularization functions
  • Reduction operations
  • Comparison operations
  • Logical operations
  • Bitwise operations
  • Tensor manipulation operations
  • Indexing operations
  • Iterators
  • Memory Hierarchy
  • Others
  • Data Types
  • Constants
• nki.collectives
  • NKI Collectives
  • Constants
• 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

This document is relevant for: Trn2, Trn3