I like the simplicity of this mod and how straightforward it is, but would it be possible to add combability with K2 so it has equipment grids and can take vehicle equipment?

Krastorio 2! I hear it is a great mod.
However, I don't feel like I can implement your idea because I've never played Krastorio 2 and don’t know a lot about it.

If it's a simple issue such as train speed, I am interested.

Ah, no worries. I was just testing your mod and other Electric Train mods. I really like this one, but Krastorio 2 has nerfs to train speed, so you need to add stuff to its equipment grid to get its acceleration up. :)

First thing to try is adding Krastorio 2 as a hidden dependency. If Krastorio adds the grid to the vanilla loco in the data stage, then your table.deepcopy will copy the grid as well. However, I think the grid is added in the data-updates stage, so you would have to add a data-updates.lua that looks for a grid on the vanilla prototype and copies it to yours as well.