Implement interface to NIR

kmheckel / spyx

Spyx: Spiking Neural Networks in JAX

https://spyx.readthedocs.io/en/latest/

MIT License

98 stars 11 forks source link

Implement interface to NIR #4

Open kmheckel opened 11 months ago

kmheckel commented 11 months ago

Create functions to load/save models to HDF5 under the Neuromorphic Intermediate Representation standard to facilitate cross-platform deployment.

https://nnir.readthedocs.io/en/latest/what.html

[x] Implement spyx.nir.to_nir()
[x] Implement spyx.nir.from_nir()
[x] Support feed-forward network import
[x] Support ConvNet import
[x] Support explicitly recurrent import
[x] Implement FFN exporting
[ ] Implement CSNN exporting
[ ] Implement RSNN exporting

kmheckel commented 10 months ago

See docs/examples/nir for notebooks documenting functionality and testing. Currently works for single neuron loading example, writing the model back is a work in progress.

kmheckel commented 10 months ago

Working on getting the weight scaling to function properly.

[x] Get weight scaling working so that loaded models function properly

kmheckel commented 10 months ago

Importing models is now supported but is likely somewhat brittle. Will require further testing. Export capabilities started but limited to FFN for now, rest is a work in progress.

kmheckel commented 6 months ago

This needs formal tests included into the library before closing this issue.