This repository contains an early version of Python bindings to Dali, the automatic differentiation framework.
With this module you can construct arbitrary mathematical models, including differentiating through control code and loops, optimize and run them using your GPU or CPU.
The bindings are light-weight, and the resulting module retains about 90%-95% of the C++ performance in most use-cases (with this performance going up for larger models and GPU intensive computations).
-
Install Dali using
homebrew
,yum
, orapt-get
. -
pip3 install dali
See examples under notebooks.