I am not able to log anything with TensorBoard using the trax library.
This is my train function (trax):
log_dir = "/some_path/logs" + datetime.d
