jaxsnn.discrete.topology

Classes

BaseState()

Base class for all neuron states across paradigms

BaseTopology()

Represents a spiking neural network (SNN) topology as a directed graph of layers.

DenseData

alias of jax.Array

Population(generator, …)

A population module for the discrete paradigm.

Projection(generator, int], …)

A projection module for the discrete paradigm.

SourcePopulation(generator, …)

A source population module for the discrete paradigm.

Topology(time_steps, dt, record_states)

Represents a discrete-time spiking neural network (SNN) topology.

Functions

jaxsnn.discrete.topology.get_logger(name: str)