Add support for compile optimizer step (#2810)

Summary:

This diff adds support for compiling the optimizer step in the train pipeline of TorchRec. The changes include:
1. Add a new argument to the TrainPipeline constructor to enable compilation of the optimizer step, and modify the optimizer step to be compiled if the argument is set to True.
2. The test_equal_to_non_pipelined_compiled test cases are also added to ensure that the compiled optimizer step produces the same results as the non-compiled version.

Differential Revision: D71031049

Author: Jingchang Zhang