Streaming variational inference: Trainer for minibatch ADVI - #710

Closed

YichengYang-Ethan wants to merge 9 commits into

pymc-devs:mainfrom

YichengYang-Ethan:streaming-trainer

Commits on Aug 8, 2026

Add streaming Trainer for minibatch ADVI
YichengYang-Ethan
committed
048868e
Pin the Trainer's batch-to-step alignment against the loader itself
YichengYang-Ethan
committed
bc97d8b
Drop the private dataloader import and the nested stream generator

YichengYang-Ethan
and
claude
committed
563dfd7
Pin the bool step count and the progressbar default

YichengYang-Ethan
and
claude
committed
4772d64
Accept any integer step count in fit, and bound the empty-source fixture

YichengYang-Ethan
and
claude
committed
2ce8a1b
Follow the merged DataLoader: total_size is the property, len is the batch count

YichengYang-Ethan
and
claude
committed
06b74b0
Drop the loud guards and the design-provenance prose

YichengYang-Ethan
and
claude
committed
ef1ded4
Pin the duck-typed loader contract the deleted isinstance guard left open

YichengYang-Ethan
and
claude
committed
c8711c8
Say once what the Trainer spares the user, and drop two subsumed tests

YichengYang-Ethan
and
claude
committed
b736877