A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

differentiable-programming discretization jax pde
1 Open Issue Need Help Last updated: Aug 1, 2025

Open Issues Need Help

View All on GitHub

AI Summary: Investigate and resolve inconsistencies in the `shift_operator` function within the `jaxdf` library. The operator produces different results when applied to `FourierSeries` and `FiniteDifferences` fields. The goal is to understand the intended behavior of the `dx` argument and ensure consistent results across different field types, potentially clarifying the documentation and making necessary code adjustments.

Complexity: 4/5
bug help wanted

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

Python
#differentiable-programming#discretization#jax#pde