allow user to specify which engine to perform computations with

[jwjeffr]

Multiple engines can be used for these computations. Let the user choose between numpy, jax, or pytorch.