Skip to content

energy

energy

Energy benchmark — per-sample energy, power, and efficiency measurement.

Classes

EnergyBenchmark

Bases: BaseBenchmark

Measures energy per token at thermal equilibrium.

Collects per-sample energy (J), power (W), throughput (tok/s), and energy-per-token (J/tok) so the stats table has real p50/min/max/std/p95 values rather than just aggregates.

Functions

ensure_registered 

ensure_registered() -> None

Register the energy benchmark if not already present.

Source code in src/openjarvis/bench/energy.py 
162
163
164
165
def ensure_registered() -> None:
    """Register the energy benchmark if not already present."""
    if not BenchmarkRegistry.contains("energy"):
        BenchmarkRegistry.register_value("energy", EnergyBenchmark)