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 | hxtorch | 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 | → |