Clean function evaluation code

- use Kokkos::UniqueToken manager to use proper context in
multi-thread context
- still works only for R^3->R^3 functions

Remains to
- check input/output types
- treat the case when the function returns an R^3 and not a list of 3 R