AstroAI Lunch Talks - May 20, 2024 - Patrick Kidger
20 May 2024 - Joshua Wing
The video can be found here: https://www.youtube.com/watch?v=jVvoJVvJGdw
Speaker: Patrick Kidger (Cradle.bio)
Title: An introduction to the JAX scientific ecosystem
Abstract: This talk provides an overview of several libraries in the open-source JAX ecosystem (such as Equinox, Diffrax, Optimistix, …) In short, we have been building an “autodifferentiable GPU-capable scipy”. These libraries offer the foundational core of tools that have made it possible for us to train neural networks (e.g. score-based diffusions for image generation), solve PDEs (e.g. a nonlinear heat equation), and smoothly handle hybridisations of the two (e.g. fit neural ODEs to scientific data). By the end of the talk, the goal is for you to be able to walk away with a slew of new modelling tools, suitable for tackling problems both in ML and in science.
This talk was jointly hosted by AstroAI and Seamless Astronomy.
Watch the talk below!