jax
96d3447e - Add mode='fan_geo_avg' to nn.initializers.variance_scaling.

Commit

1 year ago

Add mode='fan_geo_avg' to nn.initializers.variance_scaling.

References

#25766 - Add mode='fan_geo_avg' to nn.initializers.variance_scaling.

#219 - CI: 02/03/25 upstream sync

Author

carlosgmartin

carlosgmartin

Parents

Loading