einops
jaxtyping
jax
beartype==0.15.0
black
matplotlib
ruff
scipy
trimesh
pytest
tqdm
torch
torchvision
opencv-python
scikit-image
numpy