Simple Modules

Bias	A simple bias module that adds a (trainable by default) bias array (default) or scalar to the input.
Constant	Module that always returns a constant value which is optionally trainable.
Einsum	Module that implements Einsum operations between array leaves in a PyTree input.
MatMul	Module for optionally trainable matrix multiplication.