Flex tensor dev

Lars Sowarequested to merge
flex_tensor_dev into main
  • 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

Merge request reports