feat: compile training loop automatically using reactant

feat: compile training loop automatically using reactant #595