Руководство по Diffrax и JAX для решателей дифференциальных уравнений и симуляций
В туториале показано, как установить JAX, Diffrax, Equinox и Optax для GPU-ускоренных вычислений. Осваивайте адаптивные солверы ОДУ, стохастические модели и нейронные дифференциальные уравнения с примерами кода.