Zongyao Yi

HeteroConv aggregation cannot handle tuple outputs

1

### 🐛 Describe the bug I would like to use HeteroConv to wrap my custom message passing layer, which updates not just the node features but also edge attributes and...

Calculating gradient for system identification gives jaxlib.xla_extension.XlaRuntimeError: UNIMPLEMENTED: batch in most minor dimension

1

## Description I am trying to find the optimal system parameters (damping/mass etc.) to minimize the MSE between real-sim trajectories. When I tried to get the gradient from the loss...