Support structured diffusion specs in continuous-time models and enforce cd-dynamax full-diffusion constraints#214

Merged

DanWaxman merged 31 commits into

mainfrom

May 15, 2026

Commits on Apr 28, 2026

Merge branch 'main' into ml-feature-212
mattlevine22
authored
get diffusion info at init
mattlevine22
committed
give discretizer fallback if dynamicalModel init hadnt been run (support direct API by checking if bm_dim or diffusion_type are None)
mattlevine22
committed

Commits on May 10, 2026

Commits on May 11, 2026

dont use cast, just assert bm_dim is not None (fight w linter...could also push it up with a new resolvedDiffusion class)
mattlevine22
committed

Commits on May 14, 2026

Commits on May 15, 2026