Simple KS solver in JAX

The Kuramoto-Sivashinsky is a fourth-order partial differential equation that shows highly chaotic dynamics. It has become an exciting testbed for deep learning methods in physics. Here, we will code a simple Exponential Time Differencing (ETD) solver in JAX/Python. Code: github.com/Ceyron/machine-lea...
------
👉 This educational series is supported by the world-leaders in integrating machine learning and artificial intelligence with simulation and scientific computing, Pasteur Labs and Institute for Simulation Intelligence. Check out simulation.science/ for more on their pursuit of 'Nobel-Turing' technologies (arxiv.org/abs/2112.03235 ), and for partnership or career opportunities.
-------
📝 : Check out the GitHub Repository of the channel, where I upload all the handwritten notes and source-code files (contributions are very welcome): github.com/Ceyron/machine-lea...
📢 : Follow me on LinkedIn or Twitter for updates on the channel and other cool Machine Learning & Simulation stuff: / felix-koehler and / felix_m_koehler
💸 : If you want to support my work on the channel, you can become a Patreon here: / mlsim
🪙: Or you can make a one-time donation via PayPal: www.paypal.com/paypalme/Felix...
-------
⚙️ My Gear:
(Below are affiliate links to Amazon. If you decide to purchase the product or something else on Amazon through this link, I earn a small commission.)
- 🎙️ Microphone: Blue Yeti: amzn.to/3NU7OAs
- ⌨️ Logitech TKL Mechanical Keyboard: amzn.to/3JhEtwp
- 🎨 Gaomon Drawing Tablet (similar to a WACOM Tablet, but cheaper, works flawlessly under Linux): amzn.to/37katmf
- 🔌 Laptop Charger: amzn.to/3ja0imP
- 💻 My Laptop (generally I like the Dell XPS series): amzn.to/38xrABL
- 📱 My Phone: Fairphone 4 (I love the sustainability and repairability aspect of it): amzn.to/3Jr4ZmV
If I had to purchase these items again, I would probably change the following:
- 🎙️ Rode NT: amzn.to/3NUIGtw
- 💻 Framework Laptop (I do not get a commission here, but I love the vision of Framework. It will definitely be my next Ultrabook): frame.work
As an Amazon Associate I earn from qualifying purchases.
-------
Timestamps:
00:00 Intro
01:10 Exponential Time Differencing Methods & Spectral Derivatives
02:26 Domain Size as a crucial parameter
02:42 Here: the "Euler" ETD method
03:22 Simulation Algorithm for the KS equation
07:09 Imports & Constants
08:01 KS integrator class Constructor
12:19 KS integrator class Call method
15:11 Mesh & Initial Condition
16:29 Plot IC & first steps
17:40 Produce trajectory by autoregressive rollout
19:10 Visualize spatiotemporal plot
21:20 Summary
22:33 Outro

[@user-qp2ps1bk3b]

very informative! Thank you

[@MachineLearningSimulation]

Glad you enjoyed it! 😊

Well done. Thanks!

[@MachineLearningSimulation]

Thanks for watching! 😊

[@vitorheitorcardosocunha3843]

Your videos are very didactic and entertaining. Thanks! Maybe you could implement the compressible Navier-Stokes-Korteweg model using the Exponential Time differencing method? Or in Fenics? The model is also stiff and is becoming more and more popular among numerical engineers!

[@MachineLearningSimulation]

Hi, thanks for the kind words and the video suggestion :). I am not very familiar with this model but will keep it on my list for future videos. I cannot guarantee that there will be a video, but never say never :).

[@amitnayak7917]

could you do a tutorial on implementing the immersed boundary method for incompressible NS?

[@MachineLearningSimulation]

Hi,
that's an interesting topic, but I am not familiar with it. Maybe at some time there will be a video, but not in the near future. Hope you can understand 😊

[@nesslange1833]

I'd also love to see a series on BEM (Boundary Element Method) tackled by you. Hopefully in the far Future. Keep on moving.

[@lineherz]

Could you explain how you came up with the equation following the sentence "Then advance the state in time by..."? Is there any prerequisite to understand the derivation?

[@thomassavary7075]

ETD (Exponential Time Differencing) is a numerical method to solve stiff ODEs, you can have a look at indico.ictp.it/event/a08165/session/7/contribution/4/material/0/0.pdf to understand the idea behind this equation (slides 19 to 22).

[@MachineLearningSimulation]

Hi,
Thanks for the question 😊
Can you give me a time stamp for when I said that in the video? It's been some time since I uploaded it.