Examples

Welcome to the Lox examples gallery. These examples demonstrate various ways to use Lox for logging and monitoring your Python and JAX applications. All examples are self-contained and can be run independently. Each example includes a brief description of its purpose and functionality. If you have any ideas for additional examples or improvements, feel free to contribute!

🕳️ Deep Nesting: Learn how Lox helps you extract values from deep within nested function calls without polluting your function signatures and return values.

📈 Parameter Estimation: See how Lox integrates with JAX, specifically using jax.lax.scan to log intermediate states during optimization.

📺 Live Plotting: Discover how to use lox.tap with real-time plotting libraries to monitor your training or simulation progress as it happens.