[P] Training a spiking NN to produce images

• jax

  82 27,936 10.0 Python

  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  

We used the spiking neuron modules in Rockpool to build the network. These particular modules are based on Jax, which gives us compilation to GPU/TPU/CPU as well as automatic differentiation "for free". There are similar modules in Rockpool based on Torch, if you prefer to use a torch training pipeline.

• Pytorch

  336 77,783 10.0 Python

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  

We used the spiking neuron modules in Rockpool to build the network. These particular modules are based on Jax, which gives us compilation to GPU/TPU/CPU as well as automatic differentiation "for free". There are similar modules in Rockpool based on Torch, if you prefer to use a torch training pipeline.

• InfluxDB

  www.influxdata.com sponsored

  Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  

Suggest a related project