Dataclass: support pytree_node=True cached properties (!835) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge dataclass into master Jul 22, 2021

Created by: PhilipVinc

Also a spinoff from #824, so those PRs can be streamlined only on the relevant things...

This PR adds support for cached nodes that are pytree_node. I will use it in #824 to cache the mean and avoid recomputing it several times.

Dataclass: support pytree_node=True cached properties