kokkos: started implementing fix langevin