Flax Regnet (#21867) - SemanticDiff

Commit

2 years ago

Flax Regnet (#21867) * initial commit * review changes * post model PR merge * updating doc