I'm Dominik, a software engineer with a focus in Machine Learning. Contact me via Email or LinkedIn.
Custom jvps and vjps in JAX example
Custom vjps in torch example
General custom jvp and vjp tutorials here
A generic interface to create arbitrary differentiable parameter expressions in JAX and torch here
I have previously worked in algorithmic game theory research focussing on solving large partially-observable games which most notably lead to a Artificial Intelligence journal publication.
Check out my scholar profile to view my other papers.
At Pasqal, I was responsible for the development and optimization of their main torch-based computational backend pyqtorch and their JAX backend horqrux.
I was also part of the scientific software library for quantum machine learning, Qadence. To learn more, check out the qadence whitepaper.