- numpy 1.21.6
- torch 1.11.0
Experiments on the moment neurak network: working_memory_mnn.py.
$ python working_memory_mnn.py
Experiments on the multitask experiments: multitask_rnn.py.
$ python multitask_rnn.py
Experiments on the spiking neural network: snn_verification.py.
$ python snn_verification.py
Some parts of this codebase are adapted or derived from MultiTask Network and moment-neural-network under the [MIT] license. Please check their repository for more details.