parallel training

[yimuu]

Can parallel training be supported?

[phlippe]

Hi, thanks for your interest in the repo! I'm planning to integrate data parallel with pmap soon, but for now, I recommend overwriting the jitting/training function in the child trainer class with the corresponding pmap-ed versions.