[D] Is there an equivalent BigDL project for NVIDIA GPUs, which allows distributing work loads across a DL cluster with spark?

So there's this relatively new "BigDL" project" (https://bigdl.readthedocs.io/en/latest/ ), which is for Intel CPUs and Intel GPUs, but there's no mention anywhere of it working for NVIDIA GPUs.

Is there any equivalent library for NVIDIA GPUs on a spark cluster?

cluster gpus library machinelearning nvidia nvidia gpus spark