pass start/stop/step to `find_ligand_atoms` + don't always transfer to memory

[IAlibay]

Probably needs a full refactor, we should be allowing folks to analyse whatever portion of the trajectory they are looking to run.

On top of that, we probably should avoid doing a transfer to memory where we can avoid it - this is horribly memory intensive.