Based on our results, we recommend Graph Neural Networks for physics simulation workloads. The TF-GNN API has a steep initial learning curve for using its data layer, especially if there is no data ...