transformers
nltk
torch
jax
flax