torch
einops
beartype
pali-torch

palme
transformers
palm-rlhf-pytorch
tokenizers
wandb
classifier-free-guidance-pytorch>