The only ones that I know of are in https://arxiv.org/abs/2012.06684 https://arxiv.org/abs/2012.06684 (with Julia DifferentialEquations.jl and DiffTaichi), but

The only ones that I know of are in https://arxiv.org/abs/2012.06684 https://arxiv.org/abs/2012.06684 (with Julia DifferentialEquations.jl and DiffTaichi), but those are more algorithmic. There, Julia does extremely well, but the conclusion that I would draw from it is more: use a programming language with robust and differentiable differential equation solvers rather than writing simple Euler loops by hand (as this article does).