rockpool.nn.modules.torch.nir

Utilities for import and export of Rockpool torch networks to NIR. For more information see https://neuroir.org/docs/

Manuscript: https://arxiv.org/abs/2311.14641

Github: https://github.com/neuromorphs/NIR

Defines the to_nir() and from_nir() helper functions.

For more information see 🧠 ⇌ 💾 Import / export between toolchains with NIR.

Functions

from_nir(source)

Generate a rockpool model from a NIR representation

to_nir(module[, sample_data, model_name])

Convert a Rockpool module into a NIR graph for export