numpy
matplotlib
pillow
litellm
fpdf2

[cupy]
cupy-cuda12x

[jax]
jax

[tensorflow]
tensorflow

[torch]
torch
