Transfer Learning Implementation

[M-R-Schaefer]

For various applications and downstream method development, we require at least a basic transfer learning implementation.

The implementation should implement:

• a path to a pre-trained model checkpoint in the CheckpointConfig
• masking model parameters with optax mask

This is sufficient for an elementary discriminative transfer learning implementation.

[M-R-Schaefer]

Also an option to reinitialize some layers