metrics

Metrics with backend dispatch for NumPy, PyTorch, and JAX.

Submodules

jax	JAX implementation of Metrics.
numpy	NumPy implementation of Metrics.
torch	PyTorch implementation of Metrics.