Problems about GPU utilisation rate

[XHW1234]

Hi Miles,

I am running ur codes on GPU but I found the utilization rate is usually very low like around 30% or below. Did u meet similar questions before and have some suggestions? Is this because that we use Jax package? Kindly let me know.

Many thanks!

Best,
Haiwei