{"library":"diffrax","install":[{"cmd":"pip install diffrax","imports":["from diffrax import diffeqsolve","from diffrax import ODETerm","from diffrax import Tsit5","from diffrax import Euler","from diffrax import SaveAt","from diffrax import Solution"]},{"cmd":"pip install 'diffrax[cuda]' # for GPU users","imports":[]}]}