set_backend

nannos.set_backend(backend)

Set the numerical backend.

Parameters:

backend (str) – Either numpy, scipy, autograd, torch or jax.

Examples

Computing gradients

Computing gradients