Introducing Metrax: performant, efficient, and robust model evaluation metrics in JAX
At Google, as teams were migrating from TensorFlow to JAX, teams were manually reimplementing metrics that were previously provided by TensorFlow, because JAX did not have a built-in metrics library. So each team using JAX was imp...
View More View More