Comment by dionhaefner

We thought it would be cool to be able to run GPU workloads across process boundaries from JAX and torch. Train your torch networks through Julia, C++ solvers with adjoints, or any arcane autodiff potion you choose to concoct – now certified fresh without host round trips. Bon apetit!