tensorflow==2.10.0
einops
numpy