GRADSOLVE : gradients exacts rapides pour ensembles ODE sur GPU

GRADSOLVE est une librairie JAX open-source permettant de résoudre et de différentier des ensembles d'ODE de faible dimension sur GPU. Elle enregistre les étapes d'un solveur adaptatif pour rejouer une différentiation à pas fixe, offrant des gradients exacts plus rapides que les méthodes par points de contrôle. Les tests montrent un gain de performance de 2,8x en résolution et jusqu'à 14,1x pour les gradients par rapport à Diffrax.