rich
psutil
tpu-info
jax
