- Pass context into various functions. - Factor out some CUDA algorithms. - Use ATS only for update position.