Reference

API reference

Which import surface owns a name, and will reaching for it pull in torch? Those two questions cause most import errors, and this section answers them one page per kernel subsystem.

The pages are generated. mkdocstrings reads the source in src/ through griffe, which parses the code statically instead of importing it, so the reference builds with no torch, no model, and no GPU, and it cannot drift from the signatures it documents. What you read here is what the source says.

The layout follows the kernel. import reward_lens and the two pure-Python subsystems under it, reward_lens.core and reward_lens.stats, pull only numpy and scipy. Everything that touches a model lives one level down in its own subsystem and is imported from there: from reward_lens.signals import load_signal, from reward_lens.measure.battery import DirectLinearAttribution. The top level stays deliberately thin. The 1.0 names it still exposes are lazy, resolved on first access, so importing the package costs nothing you did not ask for.