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) - 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)[source]

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.

Returns:

A configured logging.Logger instance.

Return type:

Logger

kempnerforge.metrics.logger.format_metrics(step, metrics)[source]

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:
Return type:

str