Add unit test and example for differentiate_forward_vars