improvement for differentiability with heterogeneous graphs

CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia

https://carlolucibello.github.io/GraphNeuralNetworks.jl/dev/

MIT License

210 stars 47 forks source link

improvement for differentiability with heterogeneous graphs #304

Closed CarloLucibello closed 1 year ago

CarloLucibello commented 1 year ago

The limitations of Zygote in handling dictionaries is a problem for heterographs. This improves things a bit by making Dict(::Pairs...) differentiable. Predates https://github.com/JuliaDiff/ChainRules.jl/pull/648, to be removed when it is merged.