#mathematics #numerics #differentiation

num-dual

Generalized (hyper) dual numbers for the calculation of exact (partial) derivatives

5 releases (3 breaking)

0.4.1 Dec 20, 2021
0.4.0 Dec 16, 2021
0.3.0 Aug 10, 2021
0.2.0 Jul 26, 2021
0.1.0 Jul 7, 2021

#49 in Math

Download history 10/week @ 2021-09-26 3/week @ 2021-10-03 6/week @ 2021-10-10 35/week @ 2021-10-17 4/week @ 2021-10-24 20/week @ 2021-10-31 30/week @ 2021-11-07 3/week @ 2021-11-14 32/week @ 2021-11-21 41/week @ 2021-11-28 9/week @ 2021-12-05 30/week @ 2021-12-12 93/week @ 2021-12-19 15/week @ 2021-12-26 30/week @ 2022-01-02 128/week @ 2022-01-09

266 downloads per month
Used in 4 crates

MIT/Apache

220KB
3K SLoC

num-dual

crate documentation minimum rustc 1.51 documentation PyPI version

Generalized, recursive, scalar and vector (hyper) dual numbers for the automatic and exact calculation of (partial) derivatives. Including bindings for python.

Installation and Usage

Python

The python package can be installed directly from PyPI:

pip install num_dual

Rust

Add this to your Cargo.toml:

[dependencies]
num-dual = "0.4"

Example

Python

Compute the first and second derivative of a scalar-valued function.

from num_dual import derive2
import numpy as np

def f(x):
    return np.exp(x) / np.sqrt(np.sin(x)**3 + np.cos(x)**3)

x = derive2(1.5)
result = f(x)
print('f(x)    = {}'.format(result.value))
print('df/dx   = {}'.format(result.first_derivative))
print('d2f/dx2 = {}'.format(result.second_derivative))

Rust

This example defines a generic function that can be called using any (hyper) dual number and automatically calculates derivatives.

use num_dual::*;
fn f<D: DualNum<f64>>(x: D, y: D) -> D {
    x.powi(3) * y.powi(2)
}
fn main() {
    let (x, y) = (5.0, 4.0);
    // Calculate a simple derivative
    let x_dual = Dual64::from(x).derive();
    let y_dual = Dual64::from(y);
    println!("{}", f(x_dual, y_dual));                      // 2000 + [1200]ε
    // Calculate a gradient
    let xy_dual_vec = StaticVec::new_vec([x,y]).map(DualVec64::<2>::from).derive();
    println!("{}", f(xy_dual_vec[0], xy_dual_vec[1]).eps);  // [1200, 1000]
    // Calculate a Hessian
    let xy_dual2 = StaticVec::new_vec([x,y]).map(Dual2Vec64::<2>::from).derive();
    println!("{}", f(xy_dual2[0], xy_dual2[1]).v2);         // [[480, 600], [600, 250]]
    // for x=cos(t) and y=sin(t) calculate the third derivative w.r.t. t
    let t = Dual3_64::from(1.0).derive();
    println!("{}", f(t.cos(), t.sin()).v3);                 // 7.358639755305733
}

Documentation

  • You can find the documentation of the rust crate here.
  • The documentation of the python package can be found here.

Python

For the following commands to work you have to have the package installed (see: installing from source).

cd docs
make html

Open _build/html/index.html in your browser.

Dependencies

~0.1–28MB
~494K SLoC