ot.gpu ====== .. automodule:: ot.gpu