Weights and Biases integration

Support the WandbLogger.

Create a new argument class for these arguments. This logger should be supported during training but not required. We will add a new extra dependency group (wandb), required if we want to use this logger.