Investigate if RecML can use Metrax metrics

[jeffcarp]

Looks like RecML define its own set of CLU metrics here:
https://github.com/AI-Hypercomputer/RecML/blob/b58765e4da75d2080a74e82bc46abb35a392644d/recml/core/metrics/__init__.py

cc @jshin1394