I wonder if it wouldn't be possible to unify the current split between gputractography.py and tractography.py so that we have one entry point to both of them through a AFQ.tractography.track function that calls out to either of these (and maybe takes a n_gpus key-word argument to decide which to use)?