It's easy to implement a logger version for https://github.com/TuringLang/AdvancedHMC.jl/blob/master/src/sampler.jl#L63-L65 that works with TensorBoardLogger.jl or WeightsAndBiasLogger.jl.