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

Commit
1 year ago
Loading