numpy
pydantic
torch
datasets
gymnasium
matplotlib
opencv-python
scikit-learn
shapely
transformers
