I'd rename these kwargs to just `staging` and `thread_coloring`.