april-tools / cirkit

a python framework to build, learn and reason about probabilistic circuits and tensor networks
https://cirkit-docs.readthedocs.io/en/latest/
GNU General Public License v3.0
71 stars 1 forks source link

Decouple bookkeeping from layers #71

Closed lkct closed 1 year ago

lkct commented 1 year ago

Layers should only do the calculation; bookkeeping is moved to the network level.

Also, forward is now in pytorch style.