andrinr / motion-derivative

Differentiable particle simulation for drawing images

Finding initial conditions for Equation of Motion

In this repo I am trying to use Enzyme.jl and GPUArrays.jl to find the derivative of the loss of the forward solver function, with respect to its initial conditions. This repo is solely for personal educational purposes.