feat: compile training loop automatically using reactant (#969)

feat: compile training loop automatically using reactant (#969) #3490