Refactor and cleanup SolverMuJoCo.py

[adenzler-nvidia]

As discussed, a ticket to refactor SolverMuJoCo to make it easier to understand. This should also include moving the kernels to a kernels.py file.