SumProductNetworks.jl
SumProductNetworks.jl copied to clipboard
[WIP] RAT-SPNs
This branch aims to implement RAT-SPNs [1] in a Zygote / Flux compatible way.
TODO:
- [ ] Fix RAT-SPN function
- [ ] Fix region-graph data-structures
- [ ] Add Flux reqiured functions
- [ ] Testing
cc: @xukai92