Supported Primitives in NIR#
This document lists which primitives are supported by the software frameworks for conversion to and from NIR:
→: Supported for conversion from NIR←: Supported for conversion to NIR⟷: Supported for both conversion directions (to and from NIR) Please note that this list is generated automatically and may not be entirely accurate.
Primitive |
jaxsnn |
Lava |
Nengo |
Norse |
rockpool |
sinabs |
snntorch |
SpiNNaker2 |
Spyx |
|---|---|---|---|---|---|---|---|---|---|
Conv1d |
⟷ |
→ |
→ |
||||||
Conv2d |
→ |
⟷ |
⟷ |
⟷ |
→ |
⟷ |
|||
Delay |
→ |
||||||||
Flatten |
→ |
→ |
⟷ |
⟷ |
→ |
→ |
|||
Affine |
→ |
→ |
⟷ |
⟷ |
⟷ |
⟷ |
⟷ |
→ |
⟷ |
Linear |
→ |
→ |
← |
⟷ |
⟷ |
→ |
⟷ |
||
Scale |
→ |
||||||||
CubaLI |
|||||||||
CubaLIF |
→ |
→ |
⟷ |
⟷ |
⟷ |
→ |
⟷ |
||
I |
→ |
||||||||
IF |
→ |
⟷ |
⟷ |
→ |
→ |
⟷ |
|||
LI |
⟷ |
⟷ |
⟷ |
⟷ |
← |
||||
LIF |
→ |
⟷ |
⟷ |
⟷ |
⟷ |
⟷ |
→ |
⟷ |
|
AvgPool2d |
⟷ |
||||||||
SumPool2d |
→ |
→ |
⟷ |
→ |
→ |
||||
Threshold |
→ |