waxmorph.train

PyTorch training re-exports for the default waxmorph.train path.

TrainConfig([n_epochs, t_rollout, ...])	Rollout and optimizer settings for a fixed agent count.
TrainResult(model, log)	Best post-update model and its training log.
train(model, optimizer, loss_fn, *, ...[, ...])	Train through learned, mechanics, and diffusion rollout steps.