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