Print XLA Kernel Traces | Relevant env vars

[Rohanjames1997]

From jax-ml/jax#22868

I'm interested to know how which XLA kernels are being invoked at runtime for ops like matrix-multiply, add, etc.
With tensorflow/xla, the TF_CPP_MAX_VLOG_LEVEL=1 environment variable was enough to print logs that displayed the runtime backend of the emitted ops (ACL / Eigen on aarch64).
But on openxla, this variable seems to have no effect.

Is there a way to log the runtime execution flow of the low level emitted ops?

[Rohanjames1997]

TF_CPP_MAX_VLOG_LEVEL=2 TF_CPP_MIN_LOG_LEVEL=0