Last released Nov 25, 2023
Graph Neural Networks in JAX
Last released Feb 6, 2023
Approximate KNN retrieval for PyTorch
Last released Nov 30, 2020
Take it slow, compute gradients
Supported by