add timing callback (#657)

### Description
Add timing callback to log train step time in second

- [ ]  Bug fix (non-breaking change which fixes an issue)
- [x]  New feature (non-breaking change which adds functionality)
- [ ]  Refactor
- [ ]  Documentation update
- [ ]  Other (please describe):

Signed-off-by: sichu <[email protected]>

sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py

@@ -25,6 +25,7 @@
 from nemo.lightning import resume
 from nemo.lightning.pytorch import callbacks as nl_callbacks
 from nemo.lightning.pytorch.optim import MegatronOptimizerModule
+from nemo.utils.exp_manager import TimingCallback
 
 from bionemo.core.utils.dtypes import PrecisionTypes, get_autocast_dtype
 from bionemo.esm2.api import ESM2Config
@@ -217,6 +218,7 @@ def main(
         RichModelSummary(max_depth=4),
         LearningRateMonitor(),
         nl_callbacks.PreemptionCallback(),
+        TimingCallback(),
     ]
     if nsys_profiling:
         if nsys_end_step is None: