datasets
tqdm
torch
transformers
radon
einops
jaxtyping
psutil  # For memory monitoring

# If cuda
accelerate