We’ve gone a totally different way, but this is another interesting project from Erwin Coumans, on the Google Brain team, who did PyBullet. NeuralSim replaces parts of physics engines with neural networks.
https://sites.google.com/usc.edu/neuralsim
https://github.com/google-research/tiny-differentiable-simulator