Is there an example of a module designed for tools that rely on gpus? is there anything different about writing such modules?

Any of the

parabricks

modules are probably a good place to look