calibraxavitai
ADDEDCalibrax is a unified benchmarking, profiling and evaluation metrics framework for the JAX Flax NNX ecosystem, offering 140+ validated metrics, XLA FLOP counting, GPU energy monitoring, regression detection, and integration with common ML experiment tracking platforms.