Skip to content

Operations on Functions

Sebastian Benthall edited this page Sep 14, 2022 · 2 revisions

We would like to be able to perform operations on mathematical functions.

  • Derivative
  • Inverse

We are looking for a general syntax and architecture for this. It is especially challenging for functions with multiple inputs and/or outputs.

Positional arguments

jax.grad takes a 'positional' argument to determine on which dimensions it's taking the derivative.

https://jax.readthedocs.io/en/latest/_autosummary/jax.grad.html

Labeled arguments

The xarray based tools have labeled dimensions, and these could be used instead of positional (ordered) arguments.

Dolo

Dolo includes a function representation which has an argument for differentiation.

https://dolo.readthedocs.io/en/latest/model_api.html?highlight=derivative#functions

Clone this wiki locally