jax.nn.initializers.rst

jax.nn.initializers package

.. currentmodule:: jax.nn.initializers

.. automodule:: jax.nn.initializers

Initializers

This module provides common neural network layer initializers, consistent with definitions used in Keras and Sonnet.

.. autosummary::
  :toctree: _autosummary

    zeros
    ones
    uniform
    normal
    variance_scaling
    glorot_uniform
    glorot_normal
    lecun_uniform
    lecun_normal
    he_uniform
    he_normal