Spectral Difference Tools in JAX (In Progress)

This package has some tools for pseudospectral methods in JAX.

---

Installation

We can install it directly through pip

pip install git+https://github.com/jejjohnson/spectraldiffx

We also use poetry for the development environment.

git clone https://github.com/jejjohnson/spectraldiffx.git
cd spectraldiffx
conda create -n spectraldiffx python=3.11 poetry
poetry install

---

References

Software

• kernex - differentiable stencils
• FiniteDiffX - finite difference tools in JAX
• SpectralDiffX - Finite Volume Tool in Python

Algorithms

• Thiry et al, 2023 | MQGeometry 1.0 - the WENO reconstructions applied to the multilayer Quasi-Geostrophic equations, Arakawa Grid masks
• [Roullet & Gaillard, 20