jaxsnn.base.types

Classes

BasePopulation()	Base class for all neuron populations across paradigms
BaseState()	Base class for all neuron states across paradigms
Parameter	alias of jax.Array
TypeVar(name, *constraints[, bound, …])	Type variable.