2
Open
Benchmarks?
16
Would make it possible to add more typing to all classes using the namedtuple, see: https://docs.python.org/3/library/typing.html#typing.NamedTuple
Add a description, image, and links to the variational-method topic page so that developers can more easily learn about it.
To associate your repository with the variational-method topic, visit your repo's landing page and select "manage topics."
(Up for grabs for anyone who wants to do this)
Docpages generated for flax modules such as our models (RBM, RBMSymm, DenseSymm) are somehow... confused.
That's because Flax modules have a lot of methods scattered around and the attributes get put a bit everywhere.
Moreover, attributes of the model/layer itself are mixed with attributes common to all flax modules that are not really relevan