This document is relevant for: Trn2, Trn3
NKI API Reference Manual#
- nki
- nki.isa
- 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 API Common Fields
This document is relevant for: Trn2, Trn3