Wandb import

[jonaswildberger]

Moved all calls to the wandb library into try, except statements, therefore dropping the requirement to have it installed. Some further improvements were made to use it in a more generic way

[stephengreen]

Looks good. The only remaining task is to change the pip requirements.

Done