Changelog

All notable changes to this project will be documented in this file.

The format is based on Keep a Changelog, and this project adheres to Semantic Versioning.

Unreleased

Added comprehensive notebook development guidelines in docs/notebooks/README.md
get_implementations() and has_implementation() for operator dispatch introspection
Operator params guide tutorial covering inspection, override, shared params, eqx.Module as params, NN-generated stencils, and learned gradient correction

Breaking: Minimum Python version increased to 3.11 (from 3.9)
Breaking: Migrated build system from Poetry to uv for faster dependency management
Breaking: Converted from Poetry to standard pyproject.toml format (PEP 621)
Updated JAX dependency to 0.8.x (from 0.4.x) for compatibility with latest features
Updated Equinox dependency to 0.13.x (from 0.11.x)
Updated plum-dispatch dependency to 2.6.x (from 2.5.x)
shift_operator now returns only one output for FourierSeries fields (removed unnecessary second output)
Updated development dependencies: pytest (9.x), isort (7.x), bump-my-version (1.x), mkdocs plugins, and others

Removed Makefile in favor of direct uv commands
Removed debug_config dictionary and related deprecation warnings
Removed Field.is_field_complex property (deprecated in favor of .is_complex)
Removed discretization decorator from public API (deprecated)

Fixed bug in Continuous.replace_params where deprecated self.get_field was used instead of self.get_fun
Inconsistent shift_operator between FourierSeries and FiniteDifferences (#146)
Linear.eq no longer crashes when compared with non-Linear types (fixes #145)
FiniteDifferences.replace_params now preserves the accuracy field (fixes ucl-bug/jwave#224)
Import NotFoundLookupError from plum instead of plum.resolver (plum-dispatch 2.x compat)

Fixed util.get_implemented bug that was happening with the new version of plum

The Quickstart tutorial has been upgdated.
The property Field.ndim has now been moved into Field.domain.ndim, as it is fundamentally a property of the domain
The init_params function now will inherit the default parameters from its operator, to remove any source of ambiguity. This means that it should not have any default values, and an error is raised if it does.

The __about__ file has been removed, as it is redundant
The function params_map is removed, use jax.tree_util.tree_map instead.
Operators are now expected to return only their outputs, and not parameters. If you need to get the parameters of an operator use its default_params method. To minimize problems for packages relying on jaxdf, in this release the outputs of an operator are filtered to keep only the first one. This will change soon to allow the user to return arbitrary PyTrees.

JaxDF Fields are now based on equinox. In theory, this should allow to use jaxdf with all the scientific libraries for the jax ecosystem. In practice, please raise an issue when you encounter one of the inevitable bugs :)
The new operator.abstract decorator can be used to define an unimplemented operator, for specifying input arguments and docstrings.
Linear fields are now defined as equal if they have the same set of parameters and the same Domain.
Ongrid fields now have the method .add_dim(), which adds an extra tailing dimension to its parameters. This is not an in-place update: the method returns a new field.
The function jaxdf.util.get_implemented is now exposed to the user.
Added laplacian operator for FiniteDifferences fields.
JaxDF now uses standard Python logging. To set the logging level, use jaxdf.logger.set_logging_level, for example jaxdf.logger.set_logging_level("DEBUG"). The default level is INFO.
Fields have now a handy property .θ which is an alias for .params
Continuous and Linear fields now have the .is_complex property
Field and Domain are now Moduless, which are based on from equinox.Module. They are entirely equivalent to equinox.Module, but have the extra .replace method that is used to update a single field.

The property .is_field_complex is now deprecated in favor of .is_complex. Same goes for .is_real.
Field.get_field is now deprecated in favor of the __call__ method.
The @discretization decorator is deprecated, as now Fields are equinox modules. It is just not needed now, and until removed it will act as a simple pass-trough

OnGrid.from_grid now automatically adds a dimension at the end of the array for scalar fields, if needed
Added a custom operator for equinox.internal._omega._Metaω objects and Fields, which makes the library compatible with diffrax

removed jaxlib from dependencies. See https://github.com/google/jax/discussions/16380 for more information

The default_parameters function now works with custom field types
__rpow__ for OnGrid
Avoids changing parameters of OnGrid inside jax transformations
Spectral gradient for signal of even length now treats the Nyquist frequency correctly
Staggering in FiniteDifferences kernel
Incorrect behaviour for 3d staggered derivatives