OGrad is a teensy autograd engine that supports backpropogation and training. The project is an implementation Andrej Karpathy's Micrograd engine in OCaml, drawing heavily from Karpathy's Neural Networks: Zero to Hero course.
A small example of training is included in bin/main.ml.