M mishalaskin.com Training Deep Networks with Data Parallelism in Jax 122 points sebg 3 years ago 37 comments