Tutorial: 3 APIs for PDEs

[jejjohnson]

This tutorial showcases how granular/explicit one can be when defining a PDE. It will start with writing everything from scratch with loops and little by little start automating things here and there. It's a great exercise to teach people different levels of abstraction that are appropriate for different use cases.

• Scratch - Loops w/ Numba
• Functional
  • Spatial Discretisation - Slicing w/ FiniteDiffX
  • TimeStepper - Scan w/ JAX
• Automated w/ Kernex
  • Spatial Discretisation w/ kmap
  • TimeStepping w/ kscan
• Explicit
  • Spatial Discretisation w/ finitediffX
  • Time Stepping w/ diffrax
• Implicit
  • Discretisation w/ jaxdf-like
  • TimeStepping w/ jaxdf-like

---

I - Prebuilt Models

For just diving right in and using it!

---

II - Operator API

A medium level of granularity.

---

III - Functional API

The finest level of granularity available.