numpy
tensorflow
matplotlib
gym
streamlit
transformers
torch
einops
accelerate