jaxsnn.discrete.types

Classes

Any(*args, **kwargs)	Special type indicating an unconstrained type.
BaseModule(generator, GeneratorFns], …)	Base class for discrete-time network modules that follow the base protocol.
BasePopulation()	Base class for all neuron populations across paradigms
BaseState()	Base class for all neuron states across paradigms
DenseData	alias of jax.Array
GeneratorFunctions(*args, **kwargs)	Base protocol for all module function containers
Generic()	Abstract base class for generic types.
Parameter	alias of jax.Array
Population(generator, …)	A population module for the discrete paradigm.
PopulationFunctions(init, …)	Function container for population modules
Projection(generator, int], …)	A projection module for the discrete paradigm.
ProjectionFunctions(init, …)	Function container for projection modules
Protocol()	Base class for protocol classes.
SourcePopulation(generator, …)	A source population module for the discrete paradigm.
TypeVar(name, *constraints[, bound, …])	Type variable.