Workshop on Machine Learning and Automatic Differentiation in JAX for Scientific Computing

Strasbourg, France
June 8 - 12, 2026

Provisional schedule

9:00

9:30

9:45

10:30

11:00

11:45

12:30

13:00

13:45

14:00

14:30

14:45

15:00

15:30

16:00

16:30

17:00

17:30

9:00 - 10:30 Pierre Glaser

Introduction to the JAX framework

9:00 - 9:30 Patrick Kidger

The diffrax library: ODEs and neural ODEs

9:30 - 10:30

Practical session:
The diffrax library

instructor: Patrick Kidger
9:00 - 10:30 Victor Michel-Dansac & Nilo Schwencke

PINNs and other neural numerical methods

9:00 - 9:45 Hugo Frezat

Climate modeling

9:45 - 10:30 Sebastian Bodenstein & Anushan Fernando

The TORAX code for plasma physics

Coffee break

11:00 - 13:00

Practical session:
introduction to the JAX framework

instructor: Pierre Glaser
11:00 - 12:30

Practical session:
optimal control of ODEs

instructors: Killian Lutz &
Claire Schnoebelen
11:00 - 12:30

Practical session:
PINNs and other neural numerical methods

instructors: Rémi Imbach & Victor Michel-Dansac & Nilo Schwencke
11:00 - 11:45 Deniz Bezgin &
Aaron Buhendwa

The JAX-Fluids library

11:45 - 12:30 Claire Schnoebelen

Neural flow for Hamiltonian problems

Lunch

Lunch

Lunch

13:45 - 14:00

Welcome

14:00 - 15:30 Samuel Vaiter

Introduction to automatic differentiation

Coffee break

Coffee break

Coffee break

16:00 - 17:30

Practical session:
automatic differentiation

instructor: Samuel Vaiter
14:30 - 15:00 Patrick Kidger

Neural networks in JAX: the equinox library

15:00 - 16:30

Practical session:
the equinox library

instructor: Patrick Kidger
14:00 - 14:45 Roxana Sublet

Particle simulations

14:45 - 15:30 Emmanuel Franck

Differentiable cG/dG methods and numerical linear algebra

16:00 - 17:30

Practical session:
differentiable cG/dG methods and numerical linear algebra

instructors: Virgile Bertrand & Emmanuel Franck
14:00 - 15:30 Felix Koehler

Hybridization of classical solvers and neural methods

16:00 - 17:30

Practical session:
hybrid methods

instructor: Felix Koehler