GitHub

Graph Neural Network (GNN) simulator for particle dynamics in PyTorch; Extension to generate a custom gravitational N-body dataset using classical numerical integrators.

Inspired by Stanford CS224W blog post: https://medium.com/stanford-cs224w/simulating-complex-physics-with-graph-networks-step-by-step-177354cb9b05 A minimal project ...