Add validation capability for JAX train node by calling a user provided function with the current model and also providing an argument to specify how often to run this function.
Pre-Merge Checklist:
[ ] if this PR adds a feature: request merge into the latest development branch (vX.Y-dev)
[ ] if this PR fixes a bug: request merge into the latest patch branch (patch-X.Y.Z)
[ ] PR branch name is short and describes the feature/bug addressed in this PR
Add validation capability for JAX train node by calling a user provided function with the current model and also providing an argument to specify how often to run this function.
Pre-Merge Checklist:
vX.Y-dev)patch-X.Y.Z)