kempnerforge.metrics.logger

Rank-aware logging utilities for KempnerForge.

Provides a simple logging setup that:

• Only emits from rank 0 by default (avoids duplicate log lines), except a record logged with extra={"every_rank": True}, which every rank emits

• Starts every line with the local time, to the millisecond, and the rank

• Supports color-coded output for key metrics

• Integrates with Python’s standard logging module

Functions

format_metrics(step, metrics)	Format a metrics dict into a compact, color-coded log line.
get_logger(name[, rank_zero_only])	Get a logger for the given module name.

kempnerforge.metrics.logger.get_logger(name, rank_zero_only=True)

Get a logger for the given module name.

Parameters:

• name (str) – Logger name (typically __name__).

• rank_zero_only (bool) – If True, only rank 0 emits logs, except records logged with extra={"every_rank": True}, which every rank emits.

Returns:

A configured logging.Logger instance.

Return type:

Logger

kempnerforge.metrics.logger.format_metrics(step, metrics)

Format a metrics dict into a compact, color-coded log line.

Example output:

[step 1000] loss=2.34 | lr=3.00e-04 | grad_norm=1.2 | tok/s=125k | mfu=52.3% | mem=71.2/80GB

Parameters:

• step (int)

• metrics (dict[str, float | int | str])

Return type:

str