torch==2.2.2
einops==0.8.0
absl-py==0.11.0