remove wrapped activations and bump required jax version (!811) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge pv/bump into master Jul 11, 2021

Created by: PhilipVinc

Now that @wdphy16 PR has been merged, we don't need to wrap those activations anymore to make them work with complex numbers. This PR also bumps the required jax/jaxlib version to the latest.

remove wrapped activations and bump required jax version

Merge request reports