e3x.nn.initializers.InitializerFn

class e3x.nn.initializers.InitializerFn(*args, **kwargs)[source]

Bases: Protocol

Protocol for all standard initializer functions.

__call__(key, shape, dtype=<class 'jax.numpy.float64'>)[source]
Return type:

<class 'Shaped[Array, '*shape']'>