GNNs for Solving Combinatorial Optimization Problems

github.comtool

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

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   flaxmodels github.com

   tool category #jax link ↗

   Pretrained models for Jax/Flax.

2. ▲
   FLAX github.com

   tool category #python-data-science link ↗

   A neural network library for JAX that is designed for flexibility.

3. ▲
   GPJax github.com

   tool category #jax link ↗

   Gaussian processes in JAX.

   
4. ▲
   JAXopt github.com

   tool category #jax link ↗

   Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.

   
5. ▲
   NAVIX github.com

   tool category #jax link ↗

   A reimplementation of MiniGrid, a Reinforcement Learning environment, in JAX

   
6. ▲
   OTT-JAX github.com

   tool category #jax link ↗

   Optimal transport tools in JAX.

   

JAX › Models and Projects > Flax: “A JAX + Flax implementation of Combinatorial Optimization with Physics-Inspired Graph Neural Networks.”