Accelerating the inference of the Exa. TrkX pipeline