Awesome GitHub RepositoriesDifferentiable Event Simulation

torchdiffeq este o bibliotecă PyTorch pentru rezolvarea ecuațiilor diferențiale ordinare (ODE), concepută pentru a rezolva probleme cu valori inițiale și pentru a construi framework-uri de tip neural ODE. Oferă un integrator ODE diferențiabil care permite modelelor de deep learning să simuleze profunzimea continuă prin integrarea funcțiilor dinamice în timp. Biblioteca dispune de un calculator de gradient bazat pe metoda adjunctă pentru backpropagation eficient din punct de vedere al memoriei. Prin rezolvarea unui sistem adjunct augmentat înapoi în timp, calculează gradienții parametrilor fără a stoca fiecare stare intermediară a solver-ului. Proiectul acoperă integrarea numerică cu solver-e adaptive și cu pas fix, încorporând controlul erorilor și interpolarea polinomială a stării. De asemenea, suportă gestionarea diferențiabilă a evenimentelor pentru a termina solver-ele atunci când funcții scalare specifice sunt declanșate și pentru a propaga gradienții prin timpul evenimentului. Framework-ul este utilizat pentru aplicații precum fluxuri de normalizare continuă și transformarea distribuțiilor de probabilitate prin ecuații diferențiale ordinare învățate în timp continuu.