Flax > GNNs for Solving Combinatorial Optimization Problems

A JAX + Flax implementation of Combinatorial Optimization with Physics-Inspired Graph Neural Networks.