Add wandb as dependency
Signed-off-by: smajumdar <smajumdar@nvidia.com>

# What does this PR do ?

Add a one line overview of what this PR aims to accomplish.

**Collection**: [Core]

# Changelog 
- Adds wandb as a dependency specifically for core

**PR Type**:
- [ ] New Feature
- [x] Bugfix
- [ ] Documentation

If you haven't finished some of the above items you can still open "Draft" PR.

