allow for output_directory=None
The docstring of the optimize_kl class indicates that the output_directory can be set to None if the output shouldn't be saved. Nevertheless, the current implementation only partially supported this feature.
@parras do you think that this can be merged or have I overseen anything?