W willwhitney.com Parallelizing neural networks on one GPU with JAX 1 points willwhitney 5 years ago 0 comments