equinox
numpy
einops
jaxtyping
datasets
matplotlib
pydantic
optax
ipywidgets
ipympl
plotly
scipy
tyro
loguru
polars
pandas
scikit-learn
fastapi
uvicorn
gitpython
fastcore
h5py
wandb
diffusers
transformers
accelerate
peft>=0.6.0
fasttransform