I feel like weights and biases is much simpler to use, and easier to setup. Have you checked it out yet?