JAX Talk: Diffrax

What to expect?

In this session, we'll discuss “Diffrax”, a JAX-based suite of ordinary/stochastic/controlled differential equation solvers, roughly analogous to the existing torchdiffeq/torchsde/DifferentialEquations.jl packages, which are available in the PyTorch and JAX ecosystems.