Support execution without mlflow

This is a quality of life improvement when running on Jean Zay (no Internet there).

try:
  import pytorch
  MLFLOW = True
except ImportError:
  MLFLOW = False


if MLFLOW:
   mlflow.do_something()

The mlflow requirement should be set as an extra in setup.py !