torch
transformers
diffusers
jax 
jaxlib
flax