beartype
coremltools>=8.1
jaxtyping
scikit-learn
torch
wandb
tabulate
