dgl-jax-backend The supporting scripts of DGL JAX backend. The implementation could be found here: dmlc/dgl#2281