brainstate.nn module#
Neural network building blocks for constructing and training artificial and spiking neural networks: the core module system, parameter containers and transforms, a full library of layers, dynamical-systems primitives, and training utilities. Browse the reference by category below.