[This may be helpful.](https://www.machinelearningnuggets.com/how-to-use-tensorboard-in-flax/)
This may be helpful.