Flex tensor dev
- basic renaming
- Use target key to access derivative target tensor on
forward_kwargs
- ForwardDictWrapper
- Dependencies
-
keep_model_output
to select if model output is a sequence
forward_kwargs
keep_model_output
to select if model output is a sequence