flax
jax[cpu]
nodai-SHARK
orbax
transformers
torch
