Examples

This section links to reproducible, runnable Phydrax examples as public Marimo notebooks.

Wave Equation (1D)

A tutorial notebook showing PCI enforced overlays, latent-factorized modeling, and efficient JVP-based differential operators for the 1D wave equation, with comparisons to the Nvidia PhysicsNeMo implementation.

• Public notebook: wave1d

Coupled Spring-Mass ODE

A benchmark notebook for the coupled 3-DOF spring-mass system in matrix form, with normalized-time training, exact initial-condition enforcement, and comparison context against the NVIDIA PhysicsNeMo spring-mass example.

• Public notebook: spring-mass-ode