lox is a lightweight, plug-and-play logging library for JAX that seamlessly integrates with core transformations like vmap, jit, and scan. It eliminates the need for complex boilerplate by automatically routing logs alongside function outputs without cluttering the functions signature. It also supports dynamically inserting callbacks for real-time monitoring.