waxmorph.torch.train

PyTorch training primitives for shape assembly with a fixed agent count.

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.