equinox icon indicating copy to clipboard operation
equinox copied to clipboard

Published 20 hours ago • verified publisher icon patrick-kidger

Multigpu training?

Open rdilip opened this issue 1 year ago • 6 comments

So I know for a while there was an issue with equinox not playing well with pmap. With the advent of pjit (and I think now just jit), are there examples of using equinox for multigpu training?

rdilip avatar Mar 15 '23 00:03 rdilip