torch
transformers
jax
flax
altair