Deprecate `netket.nn.initializers` (!935) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge github/fork/wdphy16/complex_init into master Oct 05, 2021

Created by: wdphy16

As jax 0.2.21 has been released with the PR of complex variance scaling initializers, we can start to deprecate netket.nn.initializers in favor of jax.nn.initializers (and flax.linen.initializers just imports it).

netket.nn.initializers._complex_truncated_normal is left there because netket.nn.symmetric_linear uses it, and the corresponding function in jax is internal.

I added an argument func_name in netket.utils.deprecation.deprecated, because the deprecated function may be a functools.partial and have no __name__.

Deprecate `netket.nn.initializers`

Merge request reports