diffrax
jax>=0.4.13
equinox>=0.10.11
optax
anndata>=0.7.6
tqdm
matplotlib
seaborn
numpy
