Experimental cuda nn package