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 | ← |