Machine Learning & Simulation

33.4K subscribers

210 videos

Explaining topics of 🤖 Machine Learning & 🌊 Simulation with intuition, visualization and code. ------ Hey, welcome to my channel of explanatory videos for Ma...

Latest Videos

2D & 3D KS Equation in JAX using Exponax

On Autoregressive Neural Emulators for PDEs Talk @ IRMA Strasbourg

Transformer Neural Operator in JAX

Full Lyapunov Spectrum of Chaotic Lorenz System using JAX

Podcast with @JousefM My Channel & Research, Autodiff, Physics & Deep Learning, Education...

APEBench Talk @ Pasteur Labs Journal Club

APEBench Quickstart

Largest Lyapunov Exponent using Autodiff in JAXPython

Lyapunov Exponent in NumPy

Autoregressive Neural Emulator for Lorenz in JAX

Parallel Lorenz Simulation in JAX

Lorenz Map in NumPy

Lorenz Simulator in NumPy

Autoregressive ResNet for Kuramoto-Sivashinsky (KS) in JAX

Autodiff and Adjoints for Differentiable Physics

Reverse Mode Autodiff in Python (general compute graph)

Unrolled vs. Implicit Autodiff

Unrolled Autodiff of iterative Algorithms

UNet Tutorial in JAX

DeepONet Tutorial in JAX

Spectral Derivative in 3d using NumPy and the RFFT

NumPy.fft.rfft2 - real-valued spectral derivatives in 2D

2D Spectral Derivatives with NumPy.FFT

Softmax - PullbackvJp rule

Softmax - PushforwardJvp rule

Fourier Neural Operators (FNO) in JAX

Custom Rollout transformation in JAX (using scan)

JAX.lax.scan tutorial (for autoregressive rollout)

Upgrade the KS solver in JAX to 2nd order

Simple KS solver in JAX