jax
equinox
pydantic
numpy
jaxlib
tqdm
beartype
typing_extensions
loguru
optax
jaxtyping
pytest
ftfy
regex
statedict2pytree
jaxonlayers

[dev]
pre-commit
pytest
mkdocs
vulture

[esm]
tokenizers
huggingface_hub
transformers

[examples]
tensorflow-cpu
tensorflow_datasets
clu
flax
grain
mlflow
requests

[tests]
torch
torchvision
esm>=3.2.3
